Fixed crash
Minor fix