Fix warning in test