Fix incorrect class path