Fixed the test case breaking that I added myself in order to test travis.