restrict bug fix