Fix class names
Travis