diff --git a/qubes/core2migration.py b/qubes/core2migration.py
index 9f93391e..5fa27f77 100644
--- a/qubes/core2migration.py
+++ b/qubes/core2migration.py
@@ -149,7 +149,7 @@ class Core2Qubes(qubes.Qubes):
         vm_class_name = element.tag
         try:
             kwargs = {}
-            if vm_class_name in ["TemplateVm", "TemplateHVm"]:
+            if vm_class_name in ["QubesTemplateVm", "QubesTemplateHVm"]:
                 vm_class = qubes.vm.templatevm.TemplateVM
             elif element.get('template_qid').lower() == "none":
                 kwargs['dir_path'] = element.get('dir_path')