fix broken tests from previous commit