Fixed validation error in mainstage model.