diff --git a/pkg/cli/baseimage.go b/pkg/cli/baseimage.go
index b7d9f8d3b9..3012abeb04 100644
--- a/pkg/cli/baseimage.go
+++ b/pkg/cli/baseimage.go
@@ -98,7 +98,7 @@ func newBaseImageGenerateMatrix() *cobra.Command {
 
 				if len(validTorchVersions) > 0 {
 					found = false
-					for _, validTorchVersion := range validPythonVersions {
+					for _, validTorchVersion := range validTorchVersions {
 						if config.TorchVersion == validTorchVersion {
 							found = true
 						}