Handle cast error

lambdaclass · Sep 19, 2023 · 9273a1d · 9273a1d
1 parent 36a071c
commit 9273a1d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/hints/usort_hints.go b/pkg/hints/usort_hints.go
@@ -3,14 +3,19 @@ package hints
 import (
 	"github.com/lambdaclass/cairo-vm.go/pkg/lambdaworks"
 	"github.com/lambdaclass/cairo-vm.go/pkg/types"
+	"github.com/pkg/errors"
 )
 
 // Implements hint:
 // %{ vm_enter_scope(dict(__usort_max_size = globals().get('__usort_max_size'))) %}
 func usort_enter_scope(executionScopes *types.ExecutionScopes) error {
 	usort_max_size, err := executionScopes.Get("usort_max_size")
 
-	usort_max_size_felt := usort_max_size.(lambdaworks.Felt)
+	usort_max_size_felt, cast_ok := usort_max_size.(lambdaworks.Felt)
+
+	if !cast_ok {
+		return errors.New("Error casting usort_max_size into a Felt")
+	}
 
 	if err != nil {
 		return err