diff --git a/app/src/main/res/layout/include_home_device.xml b/app/src/main/res/layout/include_home_device.xml
index 751e6cb83..047ae6a8d 100644
--- a/app/src/main/res/layout/include_home_device.xml
+++ b/app/src/main/res/layout/include_home_device.xml
@@ -13,8 +13,6 @@
 
         <import type="com.topjohnwu.magisk.R" />
 
-        <import type="com.topjohnwu.superuser.Shell" />
-
         <variable
             name="viewModel"
             type="com.topjohnwu.magisk.ui.home.HomeViewModel" />
@@ -51,7 +49,7 @@
         <com.google.android.material.button.MaterialButton
             android:id="@+id/home_device_action_reboot"
             style="@style/WidgetFoundation.Button.Text"
-            gone="@{!Shell.rootAccess()}"
+            invisibleUnless="@{XSUKt.hasRoot}"
             popupMenu="@{R.menu.menu_reboot}"
             popupMenuOnClickListener="@{(it) -> viewModel.rebootPressed(it)}"
             android:layout_width="wrap_content"