Minor fix to method!