Fix tests broken by usage of NSHashMap instead of NNMutableWeakSet.