Fix deprecation warning during test.
Merge pull request #239 from t-bullock/deprecation-warning

Fix deprecation warning during test.