Fix for the JUnit test when added return values