Fix for junit tests.