Fixed broken test for Travis