fix bug restriction