1 files changed, 4 insertions, 2 deletions
diff --git a/src/Uri.cpp b/src/Uri.cpp
index 191aafd..73bdfc3 100644
--- a/src/Uri.cpp
+++ b/src/Uri.cpp
@@ -381,8 +381,10 @@ namespace {
             }
             ++position;
         }
-        // TODO: should also catch IN_GROUP_NOT_IPV4
-        if (state == ValidationState::IN_GROUP_COULD_BE_IPV4) {
+        if (
+            (state == ValidationState::IN_GROUP_NOT_IPV4)
+            || (state == ValidationState::IN_GROUP_COULD_BE_IPV4)
+        ) {
             // count trailing group
             ++numGroups;
         }