Fix test mode selection