Rework the tests for task 01
Oops. Fix borked tests