Fix failing scala test