Small fix to location check in restrictor and prolongator.