diff --git a/zml/module.zig b/zml/module.zig
index 81ec32d..3b7e67c 100644
--- a/zml/module.zig
+++ b/zml/module.zig
@@ -235,7 +235,7 @@ pub const CompilationContext = struct {
                 }
             }
 
-            const loaded_executable = compileModuleToPjrtExecutable(arena, self._platform, module, module_dir.?) catch |err| {
+            const loaded_executable = compileModuleToPjrtExecutable(arena, self._platform, module, module_dir) catch |err| {
                 log.err(
                     "pjrt-{s} failed to compile following valid MLIR:\n{}\n{}",
                     .{ @tagName(self._platform.target), module.op().mlirFormatter(.{}), err },