1 files changed, 8 insertions, 15 deletions
diff --git a/rigify/ui.py b/rigify/ui.py
index 827b0fc2..09257f59 100644
--- a/rigify/ui.py
+++ b/rigify/ui.py
@@ -233,23 +233,16 @@ class Sample(bpy.types.Operator):
 #import space_info  # ensure the menu is loaded first
 
 def register():
-    bpy.utils.register_module(__name__)
-
-    #bpy.types.register(DATA_PT_rigify_buttons)
-    #bpy.types.register(BONE_PT_rigify_buttons)
-    #bpy.types.register(Generate)
-    #bpy.types.register(Sample)
+    bpy.utils.register_class(DATA_PT_rigify_buttons)
+    bpy.utils.register_class(BONE_PT_rigify_buttons)
+    bpy.utils.register_class(Generate)
+    bpy.utils.register_class(Sample)
 
     #space_info.INFO_MT_armature_add.append(ui.menu_func)
-    pass
 
 
 def unregister():
-    bpy.utils.unregister_module(__name__)
-
-    #bpy.types.unregister(DATA_PT_rigify_buttons)
-    #bpy.types.unregister(BONE_PT_rigify_buttons)
-    #bpy.types.unregister(Generate)
-    #bpy.types.unregister(Sample)
-    pass
-
+    bpy.types.unregister(DATA_PT_rigify_buttons)
+    bpy.types.unregister(BONE_PT_rigify_buttons)
+    bpy.types.unregister(Generate)
+    bpy.types.unregister(Sample)