Fix what I broke in the previous commit