From b21edec6b75adc31cffee166a15446a75f5e0a3b Mon Sep 17 00:00:00 2001
From: Chang Zhe Jiet <Zhe_Jiet_CHANG_from.UFINITY@tech.gov.sg>
Date: Sat, 20 Apr 2024 18:08:03 +0800
Subject: [PATCH] use TerraformElement

---
 src/blocks/Import.ts  | 6 +++---
 src/blocks/Moved.ts   | 8 ++++----
 src/blocks/Removed.ts | 6 +++---
 3 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/blocks/Import.ts b/src/blocks/Import.ts
index d944079..4d0edec 100644
--- a/src/blocks/Import.ts
+++ b/src/blocks/Import.ts
@@ -1,12 +1,12 @@
 import { Argument, Attribute } from '../arguments';
-import { Util } from '../utils';
-import { Block, Resource } from '.';
+import { TerraformElement, Util } from '../utils';
+import { Block } from '.';
 
 /**
  * @category Block
  */
 export interface ImportArgs {
-  to: Resource;
+  to: TerraformElement | string;
   id: string;
   provider?: Argument;
 }
diff --git a/src/blocks/Moved.ts b/src/blocks/Moved.ts
index 81ad020..8744a25 100644
--- a/src/blocks/Moved.ts
+++ b/src/blocks/Moved.ts
@@ -1,13 +1,13 @@
 import { Argument, Attribute } from '../arguments';
-import { Util } from '../utils';
-import { Block, Resource } from '.';
+import { TerraformElement, Util } from '../utils';
+import { Block } from '.';
 
 /**
  * @category Block
  */
 export interface MovedArgs {
-  from: Argument;
-  to: Argument | Resource;
+  from: TerraformElement | string;
+  to: TerraformElement | string;
 }
 
 /**
diff --git a/src/blocks/Removed.ts b/src/blocks/Removed.ts
index ed9fe89..50192da 100644
--- a/src/blocks/Removed.ts
+++ b/src/blocks/Removed.ts
@@ -1,12 +1,12 @@
 import { Argument, Attribute } from '../arguments';
-import { Util } from '../utils';
-import { Block, Resource } from '.';
+import { TerraformElement, Util } from '../utils';
+import { Block } from '.';
 
 /**
  * @category Block
  */
 export interface RemovedArgs {
-  from: Resource;
+  from: TerraformElement | string;
   lifecycle: {
     destroy: boolean;
   };