Fixed Junit testing.