Add back missing references