python
diff --git a/Collapse file
‎mypy/server/update.py‎
Copy file name to clipboardExpand all lines: mypy/server/update.py
+2Lines changed: 2 additions & 0 deletions b/Collapse file
‎mypy/server/update.py‎
Copy file name to clipboardExpand all lines: mypy/server/update.py
+2Lines changed: 2 additions & 0 deletions
diff --git a/Collapse file
‎test-data/unit/fine-grained-modules.test‎
Copy file name to clipboardExpand all lines: test-data/unit/fine-grained-modules.test
+16Lines changed: 16 additions & 0 deletions b/Collapse file
‎test-data/unit/fine-grained-modules.test‎
Copy file name to clipboardExpand all lines: test-data/unit/fine-grained-modules.test
+16Lines changed: 16 additions & 0 deletions
@@ -631,6 +631,8 @@ def restore(ids: list[str]) -> None:
 
     # Find any other modules brought in by imports.
     changed_modules = [(st.id, st.xpath) for st in new_modules]
+    for m in new_modules:
+        manager.import_map[m.id] = set(m.dependencies + m.suppressed)
 
     # If there are multiple modules to process, only process one of them and return
     # the remaining ones to the caller.
 
@@ -2244,3 +2244,19 @@ undefined
 a.py:1: error: Cannot find implementation or library stub for module named "foobar"
 a.py:1: note: See https://mypy.readthedocs.io/en/stable/running_mypy.html#missing-imports
 ==
+
+[case testDaemonImportMapRefresh]
+# cmd: mypy main.py
+[file main.py]
+[file main.py.2]
+import a.b
+reveal_type(a.b.foo())
+[file a/__init__.pyi]
+[file a/b.pyi]
+import a.c
+def foo() -> a.c.C: ...
+[file a/c.pyi]
+class C: ...
+[out]
+==
+main.py:2: note: Revealed type is "a.c.C"