Fix test running
Fix test running