fix tests from #6