diff --git a/facefusion_api/core.py b/facefusion_api/core.py
index 766574c..38a35e5 100644
--- a/facefusion_api/core.py
+++ b/facefusion_api/core.py
@@ -11,12 +11,12 @@ from comfy_api_nodes.apinode_utils import bytesio_to_image_tensor, tensor_to_byt
 from httpx import Client as HttpClient, Headers
 from torch import Tensor
 
-from .types import FaceSwapperModel, NodeInputTypes
+from .types import FaceSwapperModel, InputTypes
 
 
 class SwapFaceImage:
 	@classmethod
-	def INPUT_TYPES(s) -> NodeInputTypes:
+	def INPUT_TYPES(s) -> InputTypes:
 		return\
 		{
 			'required':
@@ -83,7 +83,7 @@ class SwapFaceImage:
 
 class SwapFaceVideo:
 	@classmethod
-	def INPUT_TYPES(s) -> NodeInputTypes:
+	def INPUT_TYPES(s) -> InputTypes:
 		return\
 		{
 			'required':
diff --git a/facefusion_api/types.py b/facefusion_api/types.py
index 42fc070..5d2869c 100644
--- a/facefusion_api/types.py
+++ b/facefusion_api/types.py
@@ -1,9 +1,8 @@
 from typing import Any, Dict, Literal, TypeAlias
 
-NodeInputTypes : TypeAlias = Dict[str, Any]
+InputTypes : TypeAlias = Dict[str, Any]
+
 NodeClassMapping : TypeAlias = Dict[str, Any]
 NodeDisplayNameMapping : TypeAlias = Dict[str, str]
 
-Image : TypeAlias = Any
-
 FaceSwapperModel = Literal['hyperswap_1a_256', 'hyperswap_1b_256', 'hyperswap_1c_256']