Fixing broken tests from #10