fix broken tests due to last commit