minor bug fix: incorrect class name