slight modifications I spotted when trying to run after git push