Cleaning up my mistake