Fixes broken context manager.