Fix mistake from #131