diff --git a/tetri5/entity.py b/tetri5/entity.py
index 2fa0da0..88c6bc1 100644
--- a/tetri5/entity.py
+++ b/tetri5/entity.py
@@ -1,7 +1,6 @@
 import copy
 import random
 import pygame
-from pygame import mixer
 from typing import List, Tuple
 from types import FunctionType
 from tetri5.util import ConfigurationManager
@@ -395,7 +394,8 @@ class PieceGenerator:
             random_number = random.randint(0, 6 - len(bucket))
             bucket.append(piece_types.pop(random_number))
 
-    def _get_piece_shape(piece_number: int) -> Tuple:
+    @classmethod
+    def _get_piece_shape(cls, piece_number: int) -> Tuple:
         if piece_number == 0:
             return Piece.I_SHAPE
         if piece_number == 1:
@@ -413,7 +413,8 @@ class PieceGenerator:
 
         return None
 
-    def _get_piece_color(is_player_two: bool) -> Tuple:
+    @classmethod
+    def _get_piece_color(cls, is_player_two: bool) -> Tuple:
         random_number = random.randint(1, 3)
         player_mod = "player-1" if not is_player_two else "player-2"