fixed bug in shader code

fixed bugs in shader code