fix broken tests from last commit