Jogo em Python

Boa noite, eu estou a fazer um jogo em python, muito parecido com o "arkanoid". Estou a fazer várias telas iniciais com butões de "play" e "quit", usando o pygame, porém, quando clico num botão e entro numa tela, se voltar a essa tela não consigo clicar no mesmo botão outra vez... Alguém tem alguma dica?
Estou a usar este método:
key = pygame.key.get_pressed()
if key[pygame.K_ESCAPE]:
import pausa_nivel3