Fixed JUnit failure