Use XCTAssert instead of assert