diff --git a/map2map/models/__init__.py b/map2map/models/__init__.py
index 86f733e..f9b33c0 100644
--- a/map2map/models/__init__.py
+++ b/map2map/models/__init__.py
@@ -1,3 +1,4 @@
 from .unet import UNet
 from .vnet import VNet, VNetFat
+from .patchgan import PatchGAN
 from .conv import narrow_like