X-Git-Url: http://nitlanguage.org

diff --git a/contrib/objcwrapper/src/objc_generator.nit b/contrib/objcwrapper/src/objc_generator.nit
index db273cb..6ca50a2 100644
--- a/contrib/objcwrapper/src/objc_generator.nit
+++ b/contrib/objcwrapper/src/objc_generator.nit
@@ -63,10 +63,17 @@ redef class Sys
 	end
 end
 
+# Wrapper generator
 class CodeGenerator
+
+	# `ObjcModel` to wrap
+	var model: ObjcModel
+
 	# Generate Nit code to wrap `classes`
-	fun generate(classes: Array[ObjcClass])
+	fun generate
 	do
+		var classes = model.classes
+
 		# Open specified path or stdin
 		var file
 		var path = opt_output.value