fix broken tests due to changed functionality