Change places where we are assert equality to use XCTAssertEqual instead of XCTAssertTrue