Fixes #16

src/main/java/org/mdkt/compiler/InMemoryJavaCompiler.java

@@ -109,7 +109,7 @@ public Map<String, Class<?>> compileAll() throws Exception {
 
 		Map<String, Class<?>> classes = new HashMap<String, Class<?>>();
 		for (String className : sourceCodes.keySet()) {
-			classes.put(className, classLoader.loadClass(className));
+			classes.put(className, classLoader.findClass(className));
 		}
 		return classes;
 	}

src/test/java/org/mdkt/compiler/HelloClass.java

@@ -0,0 +1,8 @@
+package org.mdkt.compiler;
+
+public class HelloClass {
+
+    public String hello() {
+        return "hello";
+    }
+}

src/test/java/org/mdkt/compiler/InMemoryJavaCompilerTest.java

@@ -1,5 +1,6 @@
 package org.mdkt.compiler;
 
+import java.lang.reflect.Method;
 import java.util.List;
 import java.util.Map;
 
@@ -30,6 +31,29 @@ public void compile_WhenTypical() throws Exception {
 		Assert.assertEquals(1, helloClass.getDeclaredMethods().length);
 	}
 
+	@Test
+	public void compile_WhenTypicalUpdateClass() throws Exception {
+		StringBuffer sourceCode = new StringBuffer();
+
+		sourceCode.append("package org.mdkt.compiler;\n");
+		sourceCode.append("public class HelloClass {\n");
+		sourceCode.append("   public String hello() { return \"hello1\"; }");
+		sourceCode.append("}");
+
+		Class<?> oldClass = HelloClass.class;
+		Class<?> newClass = InMemoryJavaCompiler.newInstance().compile("org.mdkt.compiler.HelloClass", sourceCode.toString());
+
+		Assert.assertNotEquals(oldClass.hashCode() , newClass.hashCode());
+		Assert.assertNotEquals(oldClass.getDeclaredMethod("hello").invoke(oldClass.newInstance()) ,
+				newClass.getDeclaredMethod("hello").invoke(newClass.newInstance()));
+
+
+
+
+
+	}
+
+
 	@Test
 	public void compileAll_WhenTypical() throws Exception {
 		String cls1 = "public class A{ public B b() { return new B(); }}";
@@ -114,4 +138,6 @@ public void compile_WhenWarningsAndErrors() throws Exception {
 			throw e;
 		}
 	}
+
+
 }