Fixed errors I created in last commit