Fix #40
Minor fix