Fix blend state setting and resetting