Fix broken tests from previous commit