Correcting class reference name from JS