Minor fix - solve issue #9