Fixing junit test