Remove warning when running tests