diff --git a/tests/db_routers_unittest.py b/tests/db_routers_unittest.py
index <HASH>..<HASH> 100644
--- a/tests/db_routers_unittest.py
+++ b/tests/db_routers_unittest.py
@@ -113,7 +113,7 @@ class OQRouterTestCase(unittest.TestCase):
         For each model in the 'uiapi' schema, test for proper db routing
         for read operations.
         '''
-        classes = [Upload, Input, OqJob, OqParams, Output, ErrorMsg]
+        classes = [Upload, Input, InputSet, OqJob, OqParams, Output, ErrorMsg]
         expected_db = 'reslt_writer'
 
         self._db_for_read_helper(classes, expected_db)
@@ -123,7 +123,7 @@ class OQRouterTestCase(unittest.TestCase):
         For each model in the 'uiapi' schema, test for proper db routing
         for write operations.
         '''
-        classes = [Upload, Input, OqJob, OqParams, ErrorMsg]
+        classes = [Upload, Input, InputSet, OqJob, OqParams, ErrorMsg]
         expected_db = 'job_init'
 
         self._db_for_write_helper(classes, expected_db)
Added test to verify InputSet is routed as expected.