Many apologies.

But maybe you can use some invisible walls to restrict the entities.
Depends on your project, of course.