diff --git a/generators/python.go b/generators/python.go
index 427996c..1c88a3c 100644
--- a/generators/python.go
+++ b/generators/python.go
@@ -31,7 +31,6 @@ const pythonSrcTmpl = `
        {{- end}}) -> {{if .ReturnType }}{{ .ReturnType }}: {{ else }} None: {{ end }}
         pass
 {{- end -}}
-from boa3.builtin.interop.contract import call_contract
 from boa3.builtin.type import UInt160, UInt256, ECPoint
 from boa3.builtin import contract
 from typing import cast, Any
diff --git a/main.go b/main.go
index 3c8ab01..3e0c54c 100644
--- a/main.go
+++ b/main.go
@@ -182,6 +182,7 @@ func handleCliRun(cCtx *cli.Context) error {
 							generateSuccess = true
 						}
 					}
+					break
 				}
 			}
 			if sdkOnly && *c.ContractGenerateSdk {
@@ -189,6 +190,8 @@ func handleCliRun(cCtx *cli.Context) error {
 				if err != nil {
 					log.Fatal(err)
 				}
+				generateSuccess = true
+				break
 			}
 		}