diff --git a/code/form/WidgetAreaEditor.php b/code/form/WidgetAreaEditor.php
index 8977e87..0623f28 100644
--- a/code/form/WidgetAreaEditor.php
+++ b/code/form/WidgetAreaEditor.php
@@ -42,10 +42,13 @@ class WidgetAreaEditor extends FormField {
 		foreach($this->widgetClasses as $widgetClass) {
 			$classes = ClassInfo::subclassesFor($widgetClass);
 
-			if(count($classes) > 1) {
-				array_shift($classes);
+			if (isset($classes['Widget'])) { 
+				unset($classes['Widget']); 
+			} 
+			else if (isset($classes[0]) && $classes[0] == 'Widget') { 
+				unset($classes[0]); 
 			}
-
+			
 			foreach($classes as $class) {
 				$widgets->push(singleton($class));
 			}