using viewWillAppear here seems to work well