diff --git a/signature/converter/converter.go b/signature/converter/converter.go
index ea24e165..8d540282 100644
--- a/signature/converter/converter.go
+++ b/signature/converter/converter.go
@@ -73,7 +73,6 @@ func New(schema *signature.Schema) (*Converter, error) {
 		p.models[model.Name] = model
 		if model.Name == p.ctxName {
 			p.ctxModel = model
-			break
 		}
 	}
 	if p.ctxModel == nil {
diff --git a/signature/converter/converter_test.go b/signature/converter/converter_test.go
index 58bc0a5b..3263b5e2 100644
--- a/signature/converter/converter_test.go
+++ b/signature/converter/converter_test.go
@@ -1,3 +1,5 @@
+//go:build !integration && !generate
+
 /*
 	Copyright 2023 Loophole Labs
 	Licensed under the Apache License, Version 2.0 (the "License");