Fixed a retain cycle issue