Fix missing view-controller connection