diff --git a/modules/core/src/main/scala/stryker4s/extension/TreeExtensions.scala b/modules/core/src/main/scala/stryker4s/extension/TreeExtensions.scala
index f3ac56db7..1836a8d8f 100644
--- a/modules/core/src/main/scala/stryker4s/extension/TreeExtensions.scala
+++ b/modules/core/src/main/scala/stryker4s/extension/TreeExtensions.scala
@@ -84,19 +84,12 @@ object TreeExtensions {
   implicit final class GetMods(val tree: Tree) extends AnyVal {
     final def getMods: List[Mod] =
       tree match {
-        case mc: Defn.Class  => mc.mods
-        case mc: Defn.Trait  => mc.mods
-        case mc: Defn.Object => mc.mods
-        case mc: Defn.Def    => mc.mods
-        case mc: Defn.Val    => mc.mods
-        case mc: Defn.Var    => mc.mods
-        case mc: Defn.Type   => mc.mods
-        case mc: Term.Param  => mc.mods
-        case mc: Decl.Def    => mc.mods
-        case mc: Decl.Var    => mc.mods
-        case mc: Decl.Val    => mc.mods
-        case mc: Decl.Type   => mc.mods
-        case _               => Nil
+        case mc: Stat.WithMods           => mc.mods
+        case mc: Term.MatchLike          => mc.mods
+        case mc: Member.Param            => mc.mods
+        case mc: Ctor.Primary            => mc.mods
+        case mc: Type.FunctionParamOrArg => mc.mods
+        case _                           => Nil
       }
 
   }