Visual studio pygame.error: video system not initialized

Question

Here is my code to test pygame after installation:

import pygame

pygame.init()

screen = pygame.display.set_mode((400, 300))

done = False

while not done:

    for event in pygame.event.get():

        if event.type == pygame.QUIT:

            done = True

pygame.quit()

Here is the error:

Traceback (most recent call last):
  File "a:\Visual studio stuff.....", line 18, in <module>
    main()
  File "a:\Visual studio stuff......", line 11, in main
    for event in pygame.event.get():
                 ^^^^^^^^^^^^^^^^^^
pygame.error: video system not initialized

Ive tried to look everywhere and nothing seems to work...