Bug fix - added correct shader