Fixed error that popped up from last commit