Move code from viewDidUnload to dealloc