fixing tests :dancer: