Restore context state after drawing