diff --git a/core/src/main/scala/splain/test/AutoLift.scala b/core/src/main/scala/splain/test/AutoLift.scala index e752aff..ddd5f8f 100644 --- a/core/src/main/scala/splain/test/AutoLift.scala +++ b/core/src/main/scala/splain/test/AutoLift.scala @@ -54,7 +54,12 @@ object AutoLift { } .mkString("(", ", ", ")") - val typeStr = value.getClass.getCanonicalName.stripSuffix("$") + val typeStr = { + val canonicalName = value.getClass.getCanonicalName + if (canonicalName.endsWith("$")) + canonicalName.stripSuffix("$") + ".type" + else canonicalName + } val result = s""" |$fullPath.fromPreviousStage[$typeStr]$chunkExpr