Fix deprecation warning in tests (#365)

1 file changed