Yeah. It would surprise him that she would just disappear from his side and retreat back into the puzzle. 
I think this is where she can talk to Yami; when he finds her.