Fixed a test causing Travis to fail