repair some reference error

repair some reference error