Minor fix

Needs testing obviously