Fixed #117

Updated to add the actual reference instead of relying on the map's key.