Introduce a context manager for compiler.returnable