also fixed the comment accordingly, after the class has been renamed