Updating unit tests for Entry and Exit Gates