Fix test mode argument