Fixed bug for local execution #4