diff --git a/models/resources/storage/storage.go b/models/resources/storage/storage.go
index ece3b86..c2524d4 100644
--- a/models/resources/storage/storage.go
+++ b/models/resources/storage/storage.go
@@ -24,7 +24,7 @@ var argoType = [...]string{
 }
 
 // New creates a new instance of the StorageResource struct
-func (dma StorageType) ToArgo() string {
+func (dma StorageSize) ToArgo() string {
 	return argoType[dma]
 }