- Minor fix for weird bug