diff --git a/litex_boards/targets/avnet_aesku40.py b/litex_boards/targets/avnet_aesku40.py
index 95ced1f..872863e 100755
--- a/litex_boards/targets/avnet_aesku40.py
+++ b/litex_boards/targets/avnet_aesku40.py
@@ -115,11 +115,12 @@ def main():
         **soc_core_argdict(args)
 	)
     builder = Builder(soc, **builder_argdict(args))
-    builder.build(run=args.build)
+    if args.build:
+        builder.build()
 
     if args.load:
         prog = soc.platform.create_programmer()
-        prog.load_bitstream(os.path.join(builder.gateware_dir, soc.build_name + ".bit"))
+        prog.load_bitstream(builder.get_bitstream_filename(mode="sram"))
 
 if __name__ == "__main__":
     main()