diff --git a/sloth/annotations/container.py b/sloth/annotations/container.py
index 214934d..3eff761 100644
--- a/sloth/annotations/container.py
+++ b/sloth/annotations/container.py
@@ -1,6 +1,6 @@
 import os
 import fnmatch
-from sloth.core.exceptions import ImproperlyConfigured, NotImplementedException
+from sloth.core.exceptions import ImproperlyConfigured, NotImplementedException, InvalidArgumentException
 from sloth.core.utils import import_callable
 try:
     import cPickle as pickle