Cleaning up a little mistake