diff --git a/sys/build.rs b/sys/build.rs
index 02461a5..170a1d4 100644
--- a/sys/build.rs
+++ b/sys/build.rs
@@ -36,16 +36,15 @@ fn main() {
         println!("cargo:rustc-link-lib=cublas");
         println!("cargo:rustc-link-lib=cudart");
         println!("cargo:rustc-link-lib=cublasLt");
-        #[cfg(target_os = "windows")]
-        {
-            let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");
-            println!("cargo:rustc-link-search={}", cuda_path.display());
-        }
-        #[cfg(not(target_os = "windows"))]
-        {
-            println!("cargo:rustc-link-lib=culibos");
-            println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
-            println!("cargo:rustc-link-search=/opt/cuda/lib64");
+        cfg_if! {
+            if #[cfg(target_os = "windows")] {
+                let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");
+                println!("cargo:rustc-link-search={}", cuda_path.display());
+            } else {
+                println!("cargo:rustc-link-lib=culibos");
+                println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
+                println!("cargo:rustc-link-search=/opt/cuda/lib64");
+            }
         }
     }
     println!("cargo:rerun-if-changed=wrapper.h");