diff --git a/hack/update-codegen.sh b/hack/update-codegen.sh
index 4df77aec6c3..57841bbbeb2 100755
--- a/hack/update-codegen.sh
+++ b/hack/update-codegen.sh
@@ -87,4 +87,8 @@ grafana::codegen:run pkg
 grafana::codegen:run pkg/apimachinery
 grafana::codegen:run pkg/aggregator
 
+if [ -d "pkg/extensions/apis" ]; then
+  grafana::codegen:run pkg/extensions
+fi
+
 echo "done."