Fixed another validation error in mainstage model.
Fixed another validation error in mainstage model.