Add weak references in blocks