Fixing an error I made in previous commit