Fix 'restrict' decorator