Edited JUnit test file