diff --git a/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala b/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala
index 18054c67..292185cd 100644
--- a/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala
+++ b/spark-extension/src/main/scala/org/apache/spark/sql/blaze/BlazeCallNativeWrapper.scala
@@ -199,10 +199,11 @@ object BlazeCallNativeWrapper extends Logging {
       val tempFile = File.createTempFile("libblaze-", ".tmp")
       tempFile.deleteOnExit()
 
-      Utils.tryWithResource(classLoader.getResourceAsStream(libName)) { is =>
+      Utils.tryWithResource {
+        val is = classLoader.getResourceAsStream(libName)
         assert(is != null, s"cannot load $libName")
-        Files.copy(is, tempFile.toPath, StandardCopyOption.REPLACE_EXISTING)
-      }
+        is
+      }(Files.copy(_, tempFile.toPath, StandardCopyOption.REPLACE_EXISTING))
       System.load(tempFile.getAbsolutePath)
 
     } catch {