Updated test case

Please consider these changes to the JUnit test.