Small shader fix