Added a small modification in order to make a mock test run