fixing failing tests

hope this helps :construction_worker: