diff --git a/examples/zero_shot_prompting.py b/examples/zero_shot_prompting.py
index 34c19a9..f191390 100644
--- a/examples/zero_shot_prompting.py
+++ b/examples/zero_shot_prompting.py
@@ -1,4 +1,4 @@
-from typing import Dict, Type, Union, cast
+from typing import Dict, Type, Union
 
 import torch
 from transformers.modeling_utils import PreTrainedModel