Fixing unit test, making Travis happy.