diff --git a/README.md b/README.md
index b7a0ca5..368ade0 100644
--- a/README.md
+++ b/README.md
@@ -78,7 +78,7 @@ and a corresponding vector in numpy.ndarray given a video input data.
 ​	Input video data using local path in string or video frames in ndarray.
 
 
-**Returns**: *(list, list, Tensor)*
+**Returns**: *(list, list, torch.Tensor)*
 
 ​	A tuple of (labels, scores, features),
 which contains lists of predicted class names and corresponding scores.