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

diff --git a/src/metrics/static_types_metrics.nit b/src/metrics/static_types_metrics.nit
index b94ca13..4dd38d0 100644
--- a/src/metrics/static_types_metrics.nit
+++ b/src/metrics/static_types_metrics.nit
@@ -41,21 +41,14 @@ private class ATypeCounterVisitor
 
 	var typecount: Counter[MType]
 
-	# Get a new visitor on a classef to add type count in `typecount`.
-	init(modelbuilder: ModelBuilder, nclassdef: AClassdef, typecount: Counter[MType])
-	do
-		self.modelbuilder = modelbuilder
-		self.nclassdef = nclassdef
-		self.typecount = typecount
-	end
-
 	redef fun visit(n)
 	do
 		if n isa AAnnotation then return
 
-		if n isa AType then
+		if n isa AType then do
 			var mclassdef = self.nclassdef.mclassdef
-			var mtype = modelbuilder.resolve_mtype(mclassdef.mmodule, mclassdef, n)
+			if mclassdef == null then break
+			var mtype = modelbuilder.resolve_mtype(mclassdef, n)
 			if mtype != null then
 				self.typecount.inc(mtype)
 			end