Fix WeakMap example