Updated test to pass travis build