Fix callback problem due to weakHashMap