small shader fix