fix junit tests