diff --git a/internal/codegen/common.go b/internal/codegen/common.go
index 87dccc2..9656141 100644
--- a/internal/codegen/common.go
+++ b/internal/codegen/common.go
@@ -42,7 +42,8 @@ type nullableHelper struct {
 	ArgType      string
 }
 
-func (b *IndentStringBuilder) writeNullableHelpers(nullableHelpers core.NullableHelpers, nonNullAnnotation, nullableAnnotation string) {
+func (b *IndentStringBuilder) writeNullableHelpers(nullableHelpers core.NullableHelpers, nonNullAnnotation, nullableAnnotation string) []string {
+	imports := make([]string, 0)
 	methodTypes := []nullableHelper{
 		{nullableHelpers.Int, "Integer", "Int"},
 		{nullableHelpers.Long, "Long", "Long"},
@@ -77,7 +78,11 @@ func (b *IndentStringBuilder) writeNullableHelpers(nullableHelpers core.Nullable
 		))
 		b.WriteIndentedString(2, "var colVal = rs.getArray(col); return colVal == null ? null : Arrays.asList(as.cast(colVal.getArray()));\n")
 		b.WriteIndentedString(1, "}\n")
+
+		imports = append(imports, "java.util.List")
 	}
+
+	return imports
 }
 
 func (b *IndentStringBuilder) writeParameter(javaType core.JavaType, name, nonNullAnnotation, nullableAnnotation string) ([]string, error) {
diff --git a/internal/codegen/queries.go b/internal/codegen/queries.go
index f3d68f8..f0d0205 100644
--- a/internal/codegen/queries.go
+++ b/internal/codegen/queries.go
@@ -162,7 +162,9 @@ func BuildQueriesFile(engine string, config core.Config, queryFilename string, q
 
 	// boilerplate methods to allow for getting null primitive values
 	body.WriteString("\n")
-	body.writeNullableHelpers(nullableHelpers, nonNullAnnotation, nullableAnnotation)
+
+	imp := body.writeNullableHelpers(nullableHelpers, nonNullAnnotation, nullableAnnotation)
+	imports = append(imports, imp...)
 
 	for _, q := range queries {
 		body.WriteString("\n")