diff --git a/frontend/src/force-simulation.ts b/frontend/src/force-simulation.ts
index 9d317b6..95c77d0 100644
--- a/frontend/src/force-simulation.ts
+++ b/frontend/src/force-simulation.ts
@@ -91,10 +91,8 @@ export function forceSimulation(tasks: TasksFile, ticked: (positions: Map<string
         .force(
             "link",
             d3
-                .forceLink() // This force provides links between nodes
-                .id(function (d) {
-                    return d.id;
-                }) // This provide  the id of a node
+                .forceLink<TaskId, SimulationLinkDatum<TaskId>>() // This force provides links between nodes
+                .id(d => d.id) // This provide  the id of a node
                 .links(edges) // and this the list of links
         )
         .force("charge", d3.forceManyBody().strength(repulsionForce)) // This adds repulsion between nodes. Play with the -400 for the repulsion strength
diff --git a/frontend/src/ksp-task-grabber.ts b/frontend/src/ksp-task-grabber.ts
index c666eaf..d76bcf1 100644
--- a/frontend/src/ksp-task-grabber.ts
+++ b/frontend/src/ksp-task-grabber.ts
@@ -1,4 +1,3 @@
-import type { TaskId } from "./graph-types"
 
 export type TaskAssignmentData = {
     id: string,