Minor fix for making it run more easily