diff --git a/dataset.py b/dataset.py
index ed0f8ec..d5a6186 100644
--- a/dataset.py
+++ b/dataset.py
@@ -1,6 +1,6 @@
 from dataclasses import dataclass, field
 from enum import Enum
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 import json, argparse, os, uuid
 
 # fmt: off