junit tests fix