Added a flag for the updateStep.g