diff --git a/board/src/moves/king.rs b/board/src/moves/king.rs
index b345e2e..d1c7287 100644
--- a/board/src/moves/king.rs
+++ b/board/src/moves/king.rs
@@ -161,13 +161,13 @@ impl<'pos> Iterator for KingMoveGenerator<'pos> {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::{Position, Square};
+    use crate::{piece::Piece, Position, Square};
     use std::collections::HashSet;
 
     #[test]
     fn one_king() {
         let mut pos = Position::empty();
-        pos.place_piece(&Piece::king(Color::White), &Square::e4())
+        pos.place_piece(Piece::king(Color::White), Square::E4)
             .expect("Failed to place king on e4");
 
         let generator = KingMoveGenerator::new(&pos, Color::White);
@@ -211,7 +211,7 @@ mod tests {
     #[test]
     fn one_king_corner() {
         let mut pos = Position::empty();
-        pos.place_piece(&Piece::king(Color::White), &Square::a1())
+        pos.place_piece(Piece::king(Color::White), Square::A1)
             .expect("Failed to place king on a1");
 
         let generator = KingMoveGenerator::new(&pos, Color::White);