diff --git a/src/thumbs_up/utils/function.py b/src/thumbs_up/utils/function.py
index 885cfa6..0239a10 100644
--- a/src/thumbs_up/utils/function.py
+++ b/src/thumbs_up/utils/function.py
@@ -410,7 +410,7 @@ def predictFunctionStartType(self, ea):
         """
         # Nothing to check if there is only one type
         if not self._analyzer.hasActiveCodeTypes():
-            return self._analyzer.codeTypes()[0]
+            return self._analyzer.activeCodeTypes()[0]
         # Multiple types, now predict the right one
         sample = self.extractFunctionTypeSample(ea)
         return int(self._type_classifier.predict([sample]))