Minor fix on class name associated to extension