Bug fix #360
Minor fix