Fix bug in shader uniform assignment