Fix bug introduced in prev commit

TBR