diff --git a/StarMap.Core/ModRepository/LoadedModRepository.cs b/StarMap.Core/ModRepository/LoadedModRepository.cs
index c1e0b3a..c7c8aa5 100644
--- a/StarMap.Core/ModRepository/LoadedModRepository.cs
+++ b/StarMap.Core/ModRepository/LoadedModRepository.cs
@@ -66,9 +66,9 @@ public void LoadMod(Mod mod)
                 {
                     var attr = _registeredMethodAttributes[stringAttr];
 
-                    if (!attr.IsValidSignature(classMethod)) return;
+                    if (!attr.IsValidSignature(classMethod)) continue;
 
-                    if (attr.GetType() == typeof(StarMapModAttribute))
+                    if (attr.GetType() == typeof(StarMapImmediateLoadAttribute))
                     {
                         immediateLoadMethods.Add(classMethod);
                     }