Oops, forgot a line in the python script