diff --git a/packages/chili-three/test/testEdge.ts b/packages/chili-three/test/testEdge.ts
index 2c18eb44..a6a838f4 100644
--- a/packages/chili-three/test/testEdge.ts
+++ b/packages/chili-three/test/testEdge.ts
@@ -1,5 +1,4 @@
 import {
-    GeometryEntity,
     I18nKeys,
     ICurve,
     IDocument,
@@ -9,6 +8,7 @@ import {
     LineType,
     Matrix4,
     Orientation,
+    ParameterGeometry,
     Ray,
     Result,
     Serialized,
@@ -80,7 +80,7 @@ export class TestEdge implements IEdge {
     }
 }
 
-export class TestBody extends GeometryEntity {
+export class TestBody extends ParameterGeometry {
     display: I18nKeys = "body.line";
     constructor(
         document: IDocument,