Problemas con 3D Touch y restauración de estado

Estoy trabajando para implementar la restauración del estado en mi aplicación, sin embargo, me he topado con un problema. Recientemente agregué soporte 3D Touch a la vista de tabla, y funcionó perfectamente cada vez, excepto ahora que agregué la restauración de estado. Cuando la aplicación restaura la vista de tabla, aplicar presión a las celdas no hace nada, como si la vista de la tabla no se hubiera registrado para get una vista previa.

Tengo este código en viewDidLoad :

 override func viewDidLoad() { super.viewDidLoad() if #available(iOS 9.0, *) { if traitCollection.forceTouchCapability == UIForceTouchCapability.Available { registerForPreviewingWithDelegate(self, sourceView: view) } } } 

Traté de moverlo a viewDidAppear para ver si ayudaría, pero no hizo una diferencia.

¿Algunas ideas?

Ok, me di count finalmente. Tuve que mover registerForPreviewingWithDelegate para ver viewWillAppear y también tuve que añadir self. Al Principio.