Fix dumb typo
Fix context manager