Minor fix after merge of #160