Update travis config
Fix wrong class name