Very minor fix

Minor fix, fixes bug