diff --git a/sparsembed/__version__.py b/sparsembed/__version__.py
index 470cb66..b8eb205 100644
--- a/sparsembed/__version__.py
+++ b/sparsembed/__version__.py
@@ -1,3 +1,3 @@
-VERSION = (0, 0, 4)
+VERSION = (0, 0, 5)
 
 __version__ = ".".join(map(str, VERSION))
diff --git a/sparsembed/model/sparsembed.py b/sparsembed/model/sparsembed.py
index bf31413..447909e 100644
--- a/sparsembed/model/sparsembed.py
+++ b/sparsembed/model/sparsembed.py
@@ -1,5 +1,3 @@
-import string
-
 import torch
 from transformers import AutoModelForMaskedLM, AutoTokenizer