Oops

robertofrontado · Nov 9, 2018 · 97becca · 97becca
1 parent e4ba717
commit 97becca
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/Library/OkPagerView.swift b/Library/OkPagerView.swift
@@ -98,7 +98,7 @@ open class OkPagerView: UIView, UIPageViewControllerDataSource, UIPageViewContro
     }
 
     // Create a new View Controller and pass suitable data.
-    guard let viewController = dataSource.viewControllerAtIndex(index) else {
+    guard let viewController = dataSource?.viewControllerAtIndex(index) else {
       return nil
     }
 
@@ -190,10 +190,8 @@ open class OkPagerView: UIView, UIPageViewControllerDataSource, UIPageViewContro
   // MARK: - UIPageViewControllerDelegate
   open func pageViewController(_ pageViewController: UIPageViewController, didFinishAnimating finished: Bool, previousViewControllers: [UIViewController], transitionCompleted completed: Bool) {
     if completed {
-      if let pageVC = pageViewController.viewControllers?.last as? UIViewController {
-        if let delegate = delegate {
-          delegate.onPageSelected(pageVC, index: pageVC.view.tag)
-        }
+      if let pageVC = pageViewController.viewControllers?.last {
+        delegate?.onPageSelected(pageVC, index: pageVC.view.tag)
         // Save currentIndex
         currentIndex = pageVC.view.tag
         pageControl?.currentPage = currentIndex