Fix possible test error