Fix deprecation warning in tests