fix broken tests from #342