fix: pygame import