diff --git a/internal/build/env.go b/internal/build/env.go
index 9cfcbb69b..0854fba24 100644
--- a/internal/build/env.go
+++ b/internal/build/env.go
@@ -170,5 +170,8 @@ func applyEnvFlags(env Environment) Environment {
 	if *CronJobFlag {
 		env.IsCronJob = true
 	}
+	if *UbuntuVersionFlag != "" {
+		env.UbuntuVersion = *UbuntuVersionFlag
+	}
 	return env
 }