More efficient set restriction