Forgetting to restore context after draw