Fix failing tests

Really not sure how these were passing earlier :/