Use viewController, not controller. Wasn't specific enough.