diff --git a/internal/schema/module.go b/internal/schema/module.go
index 34342c3812..a06bd660bd 100644
--- a/internal/schema/module.go
+++ b/internal/schema/module.go
@@ -241,7 +241,7 @@ func (m *Module) ToProto() proto.Message {
 		if m.Runtime.Arch != "" {
 			runtime.Arch = &m.Runtime.Arch
 		}
-		if m.Runtime.Arch != "" {
+		if m.Runtime.Image != "" {
 			runtime.Image = &m.Runtime.Image
 		}
 	}