diff --git a/pkg/apis/pytorch/validation/validation.go b/pkg/apis/pytorch/validation/validation.go
index 0abaab921..833461af3 100644
--- a/pkg/apis/pytorch/validation/validation.go
+++ b/pkg/apis/pytorch/validation/validation.go
@@ -170,7 +170,7 @@ func ValidateBetaOnePyTorchJobSpec(c *torchv1beta1.PyTorchJobSpec) error {
 				log.Warn("Image is undefined in the container")
 				return fmt.Errorf("PyTorchJobSpec is not valid")
 			}
-			if container.Name == torchv2.DefaultContainerName {
+			if container.Name == torchv1beta1.DefaultContainerName {
 				defaultContainerPresent = true
 			}
 		}