Modified test class to reflect travis deployment