Fixed missing reference causing build to fail