diff --git a/tensorflow/core/common_runtime/shape_refiner.cc b/tensorflow/core/common_runtime/shape_refiner.cc
index 2e29ef48189..ec655b2acd0 100644
--- a/tensorflow/core/common_runtime/shape_refiner.cc
+++ b/tensorflow/core/common_runtime/shape_refiner.cc
@@ -135,7 +135,7 @@ Status ShapeRefiner::InferShapesForFunctionSubNode(
         TF_RETURN_IF_ERROR(
             outer_context->MakeShapeFromShapeProto(proto, &handle));
         copied_shapes_and_types.push_back(
-            ShapeAndType(handle, shape_and_type.dtype, shape_and_type.type));
+            ShapeAndType(handle, shape_and_type.dtype, shape_and_type.specialized_type));
       }
 
       outer_context->set_output_handle_shapes_and_types(