Fix: return the wrong reference