[FOLD] Fix bug with key in map