diff --git a/skrl/envs/wrappers/torch/__init__.py b/skrl/envs/wrappers/torch/__init__.py
index abcbff60..b4176bbd 100644
--- a/skrl/envs/wrappers/torch/__init__.py
+++ b/skrl/envs/wrappers/torch/__init__.py
@@ -104,8 +104,8 @@ def _in(values, container):
             logger.info(f"Environment wrapper: 'auto' (class: {', '.join(base_classes)})")
 
         if _in("omni.isaac.lab.envs..*", base_classes):
-            return "isaaclab-*"
-        elif _in("omni.isaac.gym.vec_env.vec_env_base.VecEnvBase", base_classes) or _in("omni.isaac.gym.vec_env.vec_env_mt.VecEnvMT", base_classes):
+            return "isaaclab"
+        elif _in("omni.isaac.gym..*", base_classes):
             return "omniverse-isaacgym"
         elif _in(["isaacgymenvs..*", "tasks..*.VecTask"], base_classes):
             return "isaacgym-preview4"  # preview 4 is the same as 3
@@ -113,7 +113,7 @@ def _in(values, container):
             return "isaacgym-preview2"
         elif _in("robosuite.environments.", base_classes):
             return "robosuite"
-        elif _in("dm_env._environment.Environment.", base_classes):
+        elif _in("dm_env..*", base_classes):
             return "dm"
         elif _in("pettingzoo.utils.env", base_classes) or _in("pettingzoo.utils.wrappers", base_classes):
             return "pettingzoo"