Fixing UnitTest

Making Travis happy (hopefully)