Fixed #55: Minor fix