Fixes warnings from #8