diff --git a/build/mobilenet-v1/build.py b/build/mobilenet-v1/build.py
index 3c3e676..c2d3821 100644
--- a/build/mobilenet-v1/build.py
+++ b/build/mobilenet-v1/build.py
@@ -45,7 +45,7 @@
 model_name = "mobilenetv1-w4a4"
 
 # which platforms to build the networks for
-zynq_platforms = ["ZCU104"]  # "ZCU104"
+zynq_platforms = ["ZCU104", "ZCU102"]
 alveo_platforms = ["U250"]  # "U50", "U200", "U280"
 platforms_to_build = zynq_platforms + alveo_platforms
 
diff --git a/build/vgg10-radioml/custom_steps.py b/build/vgg10-radioml/custom_steps.py
index 8e747f7..509efbc 100755
--- a/build/vgg10-radioml/custom_steps.py
+++ b/build/vgg10-radioml/custom_steps.py
@@ -29,7 +29,7 @@
 from finn.builder.build_dataflow_config import DataflowBuildConfig
 from qonnx.transformation.change_3d_tensors_to_4d import Change3DTo4DTensors
 from qonnx.transformation.general import GiveUniqueNodeNames
-import finn.transformation.fpgadataflow.convert_to_hw_layers as to_hls
+import finn.transformation.fpgadataflow.convert_to_hw_layers as to_hw
 import finn.transformation.streamline.absorb as absorb
 
 
@@ -40,7 +40,7 @@ def step_pre_streamline(model: ModelWrapper, cfg: DataflowBuildConfig):
 
 
 def step_convert_final_layers(model: ModelWrapper, cfg: DataflowBuildConfig):
-    model = model.transform(to_hls.InferChannelwiseLinearLayer())
-    model = model.transform(to_hls.InferLabelSelectLayer())
+    model = model.transform(to_hw.InferChannelwiseLinearLayer())
+    model = model.transform(to_hw.InferLabelSelectLayer())
     model = model.transform(GiveUniqueNodeNames())
     return model