Gehlot told to ensure ‘discipline’ amid MLAs’ revolt