Small fix

Nasty little surprise after updating the module  :)