diff --git a/SKPhotoBrowser/SKPhotoBrowser.swift b/SKPhotoBrowser/SKPhotoBrowser.swift
index 40fe739..1bf1beb 100644
--- a/SKPhotoBrowser/SKPhotoBrowser.swift
+++ b/SKPhotoBrowser/SKPhotoBrowser.swift
@@ -277,21 +277,14 @@ public class SKPhotoBrowser: UIViewController, UIScrollViewDelegate {
         pagingScrollView.frame = frameForPagingScrollView()
         pagingScrollView.contentSize = contentSizeForPagingScrollView()
         //If we open the SKPhotoBrowser from CollectionView not the first image, we have one element in visiblepages and we should to take 0 element from visibleindex but that we should to use frame we should use currentPageIndex for frame's functions.
+        // TODO: - need to fix this bug 
         if visiblePages.count > 0 {
-            if visiblePages.count == 1 {
-                let page = visiblePages[0]
-                page.frame = frameForPageAtIndex(currentPageIndex)
-                page.setMaxMinZoomScalesForCurrentBounds()
-                if page.captionView != nil {
-                    page.captionView.frame = frameForCaptionView(page.captionView, index: currentPageIndex)
-                }
-            } else {
-                let page = visiblePages[currentPageIndex]
-                page.frame = frameForPageAtIndex(currentPageIndex)
-                page.setMaxMinZoomScalesForCurrentBounds()
-                if page.captionView != nil {
-                    page.captionView.frame = frameForCaptionView(page.captionView, index: currentPageIndex)
-                }
+            let currentIndex = visiblePages.count - 1
+            let page = visiblePages[currentIndex]
+            page.frame = frameForPageAtIndex(currentPageIndex)
+            page.setMaxMinZoomScalesForCurrentBounds()
+            if page.captionView != nil {
+                page.captionView.frame = frameForCaptionView(page.captionView, index: currentIndex)
             }
         }