diff --git a/dataherald/sql_database/models/types.py b/dataherald/sql_database/models/types.py
index 941a4390..6c937249 100644
--- a/dataherald/sql_database/models/types.py
+++ b/dataherald/sql_database/models/types.py
@@ -104,7 +104,7 @@ class DatabaseConnection(BaseModel):
 
     @classmethod
     def get_dialect(cls, input_string):
-        pattern = r"([^:/]+):/+([^/]+)/([^/]+)"
+        pattern = r"([^:/]+):/+([^/]+)/?([^/]+)"
         match = re.match(pattern, input_string)
         if not match:
             raise InvalidURIFormatError(f"Invalid URI format: {input_string}")