1 files changed, 2 insertions, 0 deletions

diff --git a/drivers/base/class.c b/drivers/base/class.c
index f098881f45..8bf2ca2e56 100644
--- a/drivers/base/class.c
+++ b/drivers/base/class.c
@@ -163,6 +163,8 @@ int class_register(struct class * cls)
 void class_unregister(struct class * cls)
 {
         pr_debug("device class '%s': unregistering\n", cls->name);
+        if (cls->virtual_dir)
+                kobject_unregister(cls->virtual_dir);
         remove_class_attrs(cls);
         subsystem_unregister(&cls->subsys);
 }