Fix weakmap test

There seems to be an error in the current WeakMap test :)