Trying to fix viewWillAppear bug