Fix wrong class call, this call refer to branch