Fixed missing reference error