Fixed test running issue