Fix test running