Update to make it work with d8a13

Hi,

This module was very useful to me so here is a patch to make it work with latest version of D8 (works for me).

Rgds,

   Quentin