should probably add a XCTAssertEqual here as well.