diff --git a/server/TracySourceView.cpp b/server/TracySourceView.cpp
index 3a935c76..73c36fa3 100644
--- a/server/TracySourceView.cpp
+++ b/server/TracySourceView.cpp
@@ -52,6 +52,7 @@ void SourceView::OpenSource( const char* fileName, int line )
     m_baseAddr = 0;
     m_symAddr = 0;
     m_currentAddr = 0;
+    m_sourceFiles.clear();
 
     ParseSource( fileName, nullptr );
     assert( !m_lines.empty() );
@@ -65,6 +66,7 @@ void SourceView::OpenSymbol( const char* fileName, int line, uint64_t baseAddr,
     m_baseAddr = baseAddr;
     m_symAddr = symAddr;
     m_currentAddr = symAddr;
+    m_sourceFiles.clear();
 
     ParseSource( fileName, &worker );
     Disassemble( baseAddr, worker );
@@ -235,6 +237,18 @@ bool SourceView::Disassemble( uint64_t symAddr, const Worker& worker )
                 }
             }
             m_asm.emplace_back( AsmLine { op.address, jumpAddr, op.mnemonic, op.op_str } );
+
+            uint32_t srcline;
+            const auto srcidx = worker.GetLocationForAddress( op.address, srcline );
+            if( srcline != 0 )
+            {
+                const auto idx = srcidx.Idx();
+                auto sit = m_sourceFiles.find( idx );
+                if( sit == m_sourceFiles.end() )
+                {
+                    m_sourceFiles.emplace( idx );
+                }
+            }
         }
         cs_free( insn, cnt );
         if( !m_jumpTable.empty() )
diff --git a/server/TracySourceView.hpp b/server/TracySourceView.hpp
index d6d74491..4537a14f 100644
--- a/server/TracySourceView.hpp
+++ b/server/TracySourceView.hpp
@@ -90,6 +90,8 @@ private:
     unordered_flat_set<uint64_t> m_jumpOut;
     int m_maxJumpLevel;
     bool m_showJumps;
+
+    unordered_flat_set<uint32_t> m_sourceFiles;
 };
 
 }