Update __init__.py

revitron · Jun 18, 2020 · b429073 · b429073
1 parent 618684d
commit b429073
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/revitron/__init__.py b/revitron/__init__.py
@@ -39,8 +39,8 @@ def _(element):
     Returns:
         mixed: A Revitron element instance
     """    
-    category = Parameter(element, 'Category').getValueString()
-        
+    category = Element(element).getParameter('Category').getValueString()
+
     switcher = {
         'RVT Links': LinkRvt(element)
     }