--- a/SmallSense__JavaCompletionEngineSimple.st	Fri May 09 17:29:26 2014 +0100
+++ b/SmallSense__JavaCompletionEngineSimple.st	Tue May 13 16:34:38 2014 +0100
@@ -133,6 +133,7 @@
     | loader loaders |
 
     loaders := Set new.
+    loaders := Set new.
     loader := (class notNil and:[class isJavaClass]) ifTrue:[class classLoader] ifFalse:[JavaVM systemClassLoader].
     [ loader notNil ] whileTrue:[
         loaders add: loader.
@@ -140,13 +141,14 @@
     ].
     loaders add: nil.
 
-    JavaVM registry  classesDo:[:cls|
-        (loaders includes: cls classLoader) ifTrue:[
+    context environment allClassesDo:[:cls|
+        (cls isJavaClass and:[loaders includes: cls classLoader]) ifTrue:[
             aBlock value: cls.
         ].
     ].
 
     "Created: / 04-10-2013 / 13:10:03 / Jan Vrany <jan.vrany@fit.cvut.cz>"
+    "Modified: / 13-05-2014 / 12:38:46 / Jan Vrany <jan.vrany@fit.cvut.cz>"
 ! !
 
 !JavaCompletionEngineSimple methodsFor:'completion-individual'!