Remove tx_context context manager