Cleaned up context manager for ImpalaContext