fix: test the context managers in more detail