diff --git a/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutSmall.java b/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutSmall.java
index 045c4a93..f4782d52 100644
--- a/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutSmall.java
+++ b/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutSmall.java
@@ -23,6 +23,14 @@ class MainLayoutSmall extends MainLayoutTray {
 		}
 	}
 
+	@Override
+	protected ArrayList<SoftKey> getKeys() {
+		if (view != null && keys.isEmpty()) {
+			keys = getKeysFromContainer(view.findViewById(R.id.main_soft_keys));
+		}
+		return keys;
+	}
+
 	@Override
 	public void setDarkTheme(boolean darkEnabled) {
 		if (view == null) {
@@ -45,12 +53,4 @@ class MainLayoutSmall extends MainLayoutTray {
 		view.findViewById(R.id.main_separator_left).setBackground(separatorColor);
 		view.findViewById(R.id.main_separator_right).setBackground(separatorColor);
 	}
-
-	@Override
-	protected ArrayList<SoftKey> getKeys() {
-		if (view != null && keys.isEmpty()) {
-			keys = getKeysFromContainer(view.findViewById(R.id.main_soft_keys));
-		}
-		return keys;
-	}
 }
diff --git a/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutTray.java b/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutTray.java
index de57babb..2b68dc17 100644
--- a/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutTray.java
+++ b/app/src/main/java/io/github/sspanak/tt9/ui/main/MainLayoutTray.java
@@ -22,10 +22,8 @@ class MainLayoutTray extends BaseMainLayout {
 	}
 
 	@Override
-	public void render() {
-		getView();
-		enableClickHandlers();
-		setSoftKeysVisibility();
+	protected ArrayList<SoftKey> getKeys() {
+		return keys;
 	}
 
 	@Override
@@ -42,7 +40,12 @@ class MainLayoutTray extends BaseMainLayout {
 	}
 
 	@Override
-	protected ArrayList<SoftKey> getKeys() {
-		return keys;
+	public void render() {
+		getView();
+		enableClickHandlers();
+		setSoftKeysVisibility();
+		for (SoftKey key : getKeys()) {
+			key.render();
+		}
 	}
 }