Extracting TypeAnnotatedAstStep. Transformation is failing