Fixed the JUnit test failure. (Wow, they are already useful)