Fix test mode