BUG: fixed ref error
ENH: add refs