diff --git a/plonky2/src/gates/mod.rs b/plonky2/src/gates/mod.rs
index df65b44c..48e319ef 100644
--- a/plonky2/src/gates/mod.rs
+++ b/plonky2/src/gates/mod.rs
@@ -3,7 +3,6 @@
 
 pub mod arithmetic_base;
 pub mod arithmetic_extension;
-pub mod assert_le;
 pub mod base_sum;
 pub mod constant;
 pub mod exponentiation;
diff --git a/plonky2/src/gates/assert_le.rs b/waksman/src/gates/assert_le.rs
similarity index 96%
rename from plonky2/src/gates/assert_le.rs
rename to waksman/src/gates/assert_le.rs
index 19bff044..c67a7125 100644
--- a/plonky2/src/gates/assert_le.rs
+++ b/waksman/src/gates/assert_le.rs
@@ -1,26 +1,25 @@
 use std::marker::PhantomData;
 
+use plonky2::gates::gate::Gate;
+use plonky2::gates::packed_util::PackedEvaluableBase;
+use plonky2::gates::util::StridedConstraintConsumer;
+use plonky2::hash::hash_types::RichField;
+use plonky2::iop::ext_target::ExtensionTarget;
+use plonky2::iop::generator::{GeneratedValues, SimpleGenerator, WitnessGenerator};
+use plonky2::iop::target::Target;
+use plonky2::iop::wire::Wire;
+use plonky2::iop::witness::{PartitionWitness, Witness};
+use plonky2::plonk::circuit_builder::CircuitBuilder;
+use plonky2::plonk::plonk_common::{reduce_with_powers, reduce_with_powers_ext_circuit};
+use plonky2::plonk::vars::{
+    EvaluationTargets, EvaluationVars, EvaluationVarsBase, EvaluationVarsBaseBatch,
+    EvaluationVarsBasePacked,
+};
 use plonky2_field::extension::Extendable;
 use plonky2_field::packed::PackedField;
 use plonky2_field::types::{Field, Field64};
 use plonky2_util::{bits_u64, ceil_div_usize};
 
-use crate::gates::gate::Gate;
-use crate::gates::packed_util::PackedEvaluableBase;
-use crate::gates::util::StridedConstraintConsumer;
-use crate::hash::hash_types::RichField;
-use crate::iop::ext_target::ExtensionTarget;
-use crate::iop::generator::{GeneratedValues, SimpleGenerator, WitnessGenerator};
-use crate::iop::target::Target;
-use crate::iop::wire::Wire;
-use crate::iop::witness::{PartitionWitness, Witness};
-use crate::plonk::circuit_builder::CircuitBuilder;
-use crate::plonk::plonk_common::{reduce_with_powers, reduce_with_powers_ext_circuit};
-use crate::plonk::vars::{
-    EvaluationTargets, EvaluationVars, EvaluationVarsBase, EvaluationVarsBaseBatch,
-    EvaluationVarsBasePacked,
-};
-
 // TODO: replace/merge this gate with `ComparisonGate`.
 
 /// A gate for checking that one value is less than or equal to another.
@@ -450,6 +449,11 @@ mod tests {
     use std::marker::PhantomData;
 
     use anyhow::Result;
+    use plonky2::gates::gate::Gate;
+    use plonky2::gates::gate_testing::{test_eval_fns, test_low_degree};
+    use plonky2::hash::hash_types::HashOut;
+    use plonky2::plonk::config::{GenericConfig, PoseidonGoldilocksConfig};
+    use plonky2::plonk::vars::EvaluationVars;
     use plonky2_field::extension::quartic::QuarticExtension;
     use plonky2_field::goldilocks_field::GoldilocksField;
     use plonky2_field::types::Field;
@@ -457,11 +461,6 @@ mod tests {
     use rand::Rng;
 
     use crate::gates::assert_le::AssertLessThanGate;
-    use crate::gates::gate::Gate;
-    use crate::gates::gate_testing::{test_eval_fns, test_low_degree};
-    use crate::hash::hash_types::HashOut;
-    use crate::plonk::config::{GenericConfig, PoseidonGoldilocksConfig};
-    use crate::plonk::vars::EvaluationVars;
 
     #[test]
     fn wire_indices() {
diff --git a/waksman/src/gates/mod.rs b/waksman/src/gates/mod.rs
index 5a2a8f48..c73890b1 100644
--- a/waksman/src/gates/mod.rs
+++ b/waksman/src/gates/mod.rs
@@ -1 +1,2 @@
+pub mod assert_le;
 pub mod switch;
diff --git a/waksman/src/sorting.rs b/waksman/src/sorting.rs
index ac598dc8..010bc8b9 100644
--- a/waksman/src/sorting.rs
+++ b/waksman/src/sorting.rs
@@ -3,7 +3,6 @@ use std::marker::PhantomData;
 use itertools::izip;
 use plonky2::field::extension::Extendable;
 use plonky2::field::types::Field;
-use plonky2::gates::assert_le::AssertLessThanGate;
 use plonky2::hash::hash_types::RichField;
 use plonky2::iop::generator::{GeneratedValues, SimpleGenerator};
 use plonky2::iop::target::{BoolTarget, Target};
@@ -11,6 +10,7 @@ use plonky2::iop::witness::{PartitionWitness, Witness};
 use plonky2::plonk::circuit_builder::CircuitBuilder;
 use plonky2_util::ceil_div_usize;
 
+use crate::gates::assert_le::AssertLessThanGate;
 use crate::permutation::assert_permutation_circuit;
 
 pub struct MemoryOp<F: Field> {