Fixed something that I changed when trying to fix the last problem