Fixed funky bug in shader