1 files changed, 6 insertions, 9 deletions
diff --git a/src/screen/screen_choiceArena.c b/src/screen/screen_choiceArena.c
index ca14c0c..b12b8ec 100644
--- a/src/screen/screen_choiceArena.c
+++ b/src/screen/screen_choiceArena.c
@@ -99,23 +99,20 @@ static void eventWidgetButtonImage(void *p)
 	}
 }
 
-arenaFile_t* getChoiceArenaId()
+arenaFile_t* getChoiceArena()
 {
 	return currentArena;
 }
 
-void setChoiceArenaId(int n)
+void setChoiceArena(arenaFile_t *arenaFile)
 {
 	widget_t *widget_buttonimage;
+	int id;
 
-	if( n < 0 )
-	{
-		return;
-	}
-
-	currentArena = getArenaFile(n);
+	id = getArenaFileID(arenaFile);
+	currentArena = arenaFile;
 
-	widget_buttonimage = (widget_t *)listWidgetButtonimage->list[n];
+	widget_buttonimage = (widget_t *)listWidgetButtonimage->list[id];
 	setWidgetButtonimageActive(widget_buttonimage, TRUE);
 }