diff --git a/src/base.ts b/src/base.ts
index 4d38881..de15b0c 100644
--- a/src/base.ts
+++ b/src/base.ts
@@ -824,6 +824,7 @@ export default class Base {
       this.polygonEntity = null;
       this.outlineEntity = null;
       this.lineEntity = null;
+	  if(this.points.length <= 2) this.removeTempLine();
     } else if (this.type === 'line') {
       this.viewer.entities.remove(this.lineEntity);
     }