Fix pytest warnings

Before:

==================================== 297 passed, 11 warnings in 0.97s ====================================

After:

========================================== 297 passed in 0.98s ===========================================