bug fix
proper class name
@