Fixed issue
Minor fix