diff --git a/document/utils.go b/document/utils.go
index 7ee725b..ad797f5 100644
--- a/document/utils.go
+++ b/document/utils.go
@@ -137,7 +137,7 @@ func lastPunct(tokens []string, index int) int {
 			continue
 		}
 		if isPunct(token) {
-			return len(tokens) - index
+			return i - index
 		}
 	}
 	return -1