Fix TypeError on line 108, in run