Fix retain cycles.

Fix retain cycles.