         XCTAssertTrue(mainContext.saveContextAndWait(nil))
 
         backgroundContext.performBlockAndWait() {
-            if let author = Author.allAuthorsInContext(backgroundContext).first, lastName = author.lastName {
+            if let author = backgroundContext.objectWithID(authorID) as? Author, lastName = author.lastName {
                 XCTAssertEqual(lastName, "Blah", "Last name should have been updated here")
             } else {
                 XCTFail()


        let author = Author.newAuthorInContext(mainContext)
        XCTAssertTrue(mainContext.saveContextAndWait(nil))

        // Update Test

        backgroundContext.performBlockAndWait {
            if let author = Author.allAuthorsInContext(backgroundContext).first {
                author.firstName = "Joe"
            } else {
                XCTFail()

    override func setUp() {
        super.setUp()

        let ex1 = expectationWithDescription("callback")
        stack = SharedCoordinatorStack(modelName: "TestModel", inBundle: NSBundle(forClass: SharedCoordinatorStackTests.self)) { (success, error) in
            XCTAssertTrue(success)
            ex1.fulfill()
        }
        waitForExpectationsWithTimeout(10, handler: nil)
    }

