diff --git a/careless/models/priors/wilson.py b/careless/models/priors/wilson.py
index 90cb4bc..8179c0b 100644
--- a/careless/models/priors/wilson.py
+++ b/careless/models/priors/wilson.py
@@ -6,7 +6,6 @@
 from tensorflow_probability import bijectors as tfb
 import tensorflow as tf
 import numpy as np
-from tensorflow.python.framework.errors_impl import InvalidArgumentError
 
 
 class Centric(tfd.HalfNormal, Prior):