fixed JUnit errors