[TASK] Add class mapping for 4.x class name when on 6.0

Fixes: #1