minor change in frag shader