fixes failing tests

good work @rockbot! :mag: