diff --git a/socialforce/__init__.py b/socialforce/__init__.py
index 907f3a3..ed03903 100644
--- a/socialforce/__init__.py
+++ b/socialforce/__init__.py
@@ -1,6 +1,6 @@
 """PyTorch implementation of the Social Force model."""
 
-__version__ = '0.2.1'
+__version__ = '0.2.2'
 
 from .field_of_view import FieldOfView
 from .trainer import Trainer