XCTest code fix