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

diff --git a/src/nituml.nit b/src/nituml.nit
index 330370f..2723a8b 100644
--- a/src/nituml.nit
+++ b/src/nituml.nit
@@ -41,12 +41,12 @@ private class UMLPhase
 	super Phase
 	redef fun process_mainmodule(mainmodule, mmodules)
 	do
-		var view = new ModelView(mainmodule.model)
+		var filters = new ModelFilter
 		if not toolcontext.opt_privacy.value then
-			view.min_visibility = protected_visibility
+			filters.min_visibility = protected_visibility
 		end
 
-		var d = new UMLModel(view, mainmodule)
+		var d = new UMLModel(toolcontext.modelbuilder.model, mainmodule, filters)
 		if toolcontext.opt_gen.value == 0 then
 			print d.generate_class_uml.write_to_string
 		else if toolcontext.opt_gen.value == 1 then