Fix command restriction