1 files changed, 3 insertions, 1 deletions
diff --git a/src/net_types.py b/src/net_types.py
index bb52d50..14925c8 100644
--- a/src/net_types.py
+++ b/src/net_types.py
@@ -1,5 +1,7 @@
 import classification
+import counting_small
 
 types = {
-    'classification': classification
+    'classification': classification,
+    'counting-small': counting_small
 }