diff --git a/TetrisDiffingCompetition/Tetris/Tetris.swift b/TetrisDiffingCompetition/Tetris/Tetris.swift
index 8bb7740..269e4b2 100644
--- a/TetrisDiffingCompetition/Tetris/Tetris.swift
+++ b/TetrisDiffingCompetition/Tetris/Tetris.swift
@@ -237,7 +237,7 @@ class TetrisGame {
     }
 
     func spawn() -> Bool {
-        let tetromino = Tetromino.allCases.randomPick()
+        let tetromino = Tetromino.allCases.randomElement()!
 
         currentShape = blockFactory.shape(from: tetromino.template)
         x = (tetrisColumns - tetromino.template.count) / 2
@@ -306,10 +306,3 @@ class TetrisGame {
         return false
     }
 }
-
-private extension Array {
-    func randomPick() -> Element {
-        let index = Int(arc4random_uniform(UInt32(count)))
-        return self[index]
-    }
-}