fixing broken test from #150