Fixed bug in the JU shader