diff --git a/colormod.py b/colormod.py
index a6086b7..7894799 100644
--- a/colormod.py
+++ b/colormod.py
@@ -1,5 +1,4 @@
 import os
-import png
 import json
 import torch
 import numpy as np