Fix docs rendering for classes using lazy import (#651)

Fix docs rendering for classes using lazy import (#651) #238