Small fix

Nasty little surprise after updating the module  : )