diff --git a/lib/base.dart b/lib/base.dart
index 1ae1264..c309385 100644
--- a/lib/base.dart
+++ b/lib/base.dart
@@ -22,11 +22,9 @@ class _BaseState extends State {
Widget build(BuildContext context) {
final serversProvider = Provider.of(context);
- // List screens = serversProvider.selectedServer != null
- // ? screensServerConnected
- // : screensSelectServer;
-
- List screens = screensSelectServer;
+ List screens = serversProvider.selectedServer != null
+ ? screensServerConnected
+ : screensSelectServer;
if (selectedScreen > screens.length-1) {
setState(() => selectedScreen = 0);