Added essential line of code