fix broken tests from #342

the code speaks for itself ;)