Fixed a bug with the displacement