Fix warning when running test