Attempt to fix unit test for TravisCI.