Fix mismatch between class name and docs