diff --git a/SKPhotoBrowser/SKPhotoBrowser.swift b/SKPhotoBrowser/SKPhotoBrowser.swift
index a159460..aa9b3c9 100644
--- a/SKPhotoBrowser/SKPhotoBrowser.swift
+++ b/SKPhotoBrowser/SKPhotoBrowser.swift
@@ -203,19 +203,21 @@ public class SKPhotoBrowser: UIViewController, UIScrollViewDelegate {
         setup()
     }
     
-    public convenience init(photos: [SKPhotoProtocol]) {
+    public convenience init(photos: [AnyObject]) {
         self.init(nibName: nil, bundle: nil)
-        for photo in photos {
+        let picutres = photos.flatMap {$0 as? SKPhotoProtocol}
+        for photo in picutres {
             photo.checkCache()
             self.photos.append(photo)
         }
     }
     
-    public convenience init(originImage: UIImage, photos: [SKPhotoProtocol], animatedFromView: UIView) {
+    public convenience init(originImage: UIImage, photos: [AnyObject], animatedFromView: UIView) {
         self.init(nibName: nil, bundle: nil)
         self.senderOriginImage = originImage
         self.senderViewForAnimation = animatedFromView
-        for photo in photos {
+        let picutres = photos.flatMap {$0 as? SKPhotoProtocol}
+        for photo in picutres {
             photo.checkCache()
             self.photos.append(photo)
         }
diff --git a/SKPhotoBrowserTests/SKPhotoBrowserTests.swift b/SKPhotoBrowserTests/SKPhotoBrowserTests.swift
index 7ee10a9..721497b 100644
--- a/SKPhotoBrowserTests/SKPhotoBrowserTests.swift
+++ b/SKPhotoBrowserTests/SKPhotoBrowserTests.swift
@@ -7,6 +7,14 @@
 //
 
 import XCTest
+@testable import SKPhotoBrowser
+
+
+class FakeSKPhotoBrowser : SKPhotoBrowser {
+    override func setup () {
+        
+    }
+}
 
 class SKPhotoBrowserTests: XCTestCase {
     
@@ -19,12 +27,22 @@ class SKPhotoBrowserTests: XCTestCase {
         // Put teardown code here. This method is called after the invocation of each test method in the class.
         super.tearDown()
     }
+
     
-    func testExample() {
-        // This is an example of a functional test case.
-        // Use XCTAssert and related functions to verify your tests produce the correct results.
+    func testSKPhotoArray() {
+        var images = [SKPhoto]()
+        let photo = SKPhoto.photoWithImage(UIImage())// add some UIImage
+        images.append(photo)
+        let _ = FakeSKPhotoBrowser(photos: images)
     }
     
+    func testSKLocalPhotoArray() {
+        var images = [SKLocalPhoto]()
+        let photo = SKLocalPhoto.photoWithImageURL("")
+        images.append(photo)
+        let _ = FakeSKPhotoBrowser(photos: images)
+    }
+
     func testPerformanceExample() {
         // This is an example of a performance test case.
         self.measureBlock {