Fixed missing references issue