Fix to run locally