diff --git a/sunscreen_tfhe/src/ops/ciphertext/glwe_ciphertext_ops.rs b/sunscreen_tfhe/src/ops/ciphertext/glwe_ciphertext_ops.rs
index d0cfe859b..34864b97a 100644
--- a/sunscreen_tfhe/src/ops/ciphertext/glwe_ciphertext_ops.rs
+++ b/sunscreen_tfhe/src/ops/ciphertext/glwe_ciphertext_ops.rs
@@ -244,7 +244,6 @@ mod tests {
     use crate::{
         entities::{GgswCiphertext, LweCiphertext, Polynomial},
         high_level::*,
-        high_level::{keygen, TEST_GLWE_DEF_1},
         ops::encryption::{
             decrypt_ggsw_ciphertext, encrypt_ggsw_ciphertext, encrypt_glwe_ciphertext_secret,
             trivially_encrypt_glwe_ciphertext,
diff --git a/sunscreen_tfhe/src/params.rs b/sunscreen_tfhe/src/params.rs
index dc692a5b3..4d58fa568 100644
--- a/sunscreen_tfhe/src/params.rs
+++ b/sunscreen_tfhe/src/params.rs
@@ -5,9 +5,12 @@ use crate::TorusOps;
 
 use sunscreen_math::security::lwe_std_to_security_level;
 
-trait SecurityLevel {
+/// A trait for asserting properties about the security parameter.
+pub trait SecurityLevel {
+    /// Return the security level.
     fn security_level(&self) -> f64;
 
+    /// Assert the security level matches the given value.
     fn assert_security_level(&self, specified_security_level: usize) {
         // Our security level should be within 0.5 bits of the specified
         // security level (so +- 0.25 of the desired level).