Change assert to XCTAssertTrue