Quick update to script Fix line 176-177