remove hard coded class name[bug fix]