diff --git a/config.scm b/config.scm
index b1e5db7..6fad241 100644
--- a/config.scm
+++ b/config.scm
@@ -2,6 +2,7 @@
 
 (use-modules
  (gnu)
+ (gnu build file-systems)
  (gnu image)
  (gnu services authentication)
  (gnu services avahi)
@@ -44,13 +45,13 @@
   (cons*
    (file-system
     (device
-     (partition-label "EFI System Partition"))
+     (find-partition-by-label "EFI System Partition"))
     #;(device "/dev/disk/by-partlabel/EFI\\x20System\\x20Partition")
     (mount-point "/boot/efi")
     (type "vfat"))
    (file-system
     (device
-     (partition-label "guix-root"))
+     (find-partition-by-label "guix-root"))
     #;(device "/dev/disk/by-partlabel/guix-root")
     (mount-point "/")
     (type "xfs"))
@@ -59,7 +60,7 @@
   (list
    (swap-space
     (target
-     (partition-label "swap"))
+     (find-partition-by-label "swap"))
     #;(target "/dev/disk/by-partlabel/swap")
     (discard? #t))))
  (users