Convert back to XCTest