diff --git a/packages/jaeger-ui/src/model/trace-viewer.js b/packages/jaeger-ui/src/model/trace-viewer.tsx
similarity index 93%
rename from packages/jaeger-ui/src/model/trace-viewer.js
rename to packages/jaeger-ui/src/model/trace-viewer.tsx
index 98c4593a65..8f417198e3 100644
--- a/packages/jaeger-ui/src/model/trace-viewer.js
+++ b/packages/jaeger-ui/src/model/trace-viewer.tsx
@@ -12,8 +12,10 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+import { Span } from '../types/trace';
+
 // eslint-disable-next-line import/prefer-default-export
-export function getTraceName(spans) {
+export function getTraceName(spans: Span[]) {
   const allSpanIDs = spans.map(sp => sp.spanID);
   const rootSpan = spans
     .filter(sp => {
diff --git a/packages/jaeger-ui/tsconfig.lint.json b/packages/jaeger-ui/tsconfig.lint.json
index 738684b0ab..562e1172d2 100644
--- a/packages/jaeger-ui/tsconfig.lint.json
+++ b/packages/jaeger-ui/tsconfig.lint.json
@@ -30,7 +30,6 @@
     "src/api/jaeger.js",
     "src/components/SearchTracePage/SearchResults/ResultItem.markers.js",
     "src/model/order-by.js",
-    "src/model/trace-viewer.js",
     "src/selectors/process.js",
     "src/selectors/span.js",
     "src/selectors/trace.js",