bug fix old class name