Fixed small shader porting issue