Restriction logic now works