diff --git a/deploy.py b/deploy.py
index 1f000484cb05f608b096047b180fda3996ee99bc..deb0126de2ce4a18614e07375a10797c9089fa9a 100755
--- a/deploy.py
+++ b/deploy.py
@@ -28,6 +28,7 @@ import subprocess
 import sys
 
 import colorama
+from six.moves import input
 from tockloader import tab
 from tockloader import tbfh
 from tockloader import tockloader as loader
@@ -292,7 +293,7 @@ class OpenSKInstaller:
       # Need to update
       self.checked_command_output(
           ["rustup", "install", target_toolchain_fullstring])
-    targets = set([x.arch for _, x in SUPPORTED_BOARDS.items()])
+    targets = {x.arch for x in SUPPORTED_BOARDS.values()}
     for arch in targets:
       self.checked_command_output(["rustup", "target", "add", arch])
     info("Rust toolchain up-to-date")