Help me in solving TODOLIST problem

help
,
1

My issue

My code

def solution(li):
    count = 0
    for i in li:
        if i >= 1000:
            count = count + 1
    return count 

def main():
    t = int(input())
    for _ in range(t):
        n = int(input())
        li = list(map(int,input().split()))
        sol = solution(li)
        print(sol)


if __name__ == "__main__":
    main()

Problem Link: TODOLIST Problem - CodeChef