Fix wrong class