Should the test for the port be on the `InPort` header instead of `Switch`?