1 files changed, 10 insertions, 1 deletions
diff --git a/src/engine/moves/knight.c b/src/engine/moves/knight.c
index 5f5f9ad..20d072f 100644
--- a/src/engine/moves/knight.c
+++ b/src/engine/moves/knight.c
@@ -79,11 +79,14 @@ void get_knight_moves(moves_t* moves, position_t position) {
 
   bitboard_t friendly_knights;
   bitboard_t friendly_pieces;
+  bitboard_t enemy_pieces;
   if (position.turn == WHITE_TURN) {
     friendly_pieces = whites(position);
+    enemy_pieces = blacks(position);
     friendly_knights = position.bitboards[WHITE_KNIGHT];
   } else {
     friendly_pieces = blacks(position);
+    enemy_pieces = whites(position);
     friendly_knights = position.bitboards[BLACK_KNIGHT];
   }
 
@@ -95,7 +98,13 @@ void get_knight_moves(moves_t* moves, position_t position) {
     while (ATTACK_MASK) {
       int to = __builtin_ctzll(ATTACK_MASK);
       ATTACK_MASK &= ATTACK_MASK - 1;
-      add_move(moves, from, to);
+      int capture = 0;
+      add_move(
+        moves,
+        from,
+        to,
+        .flags = ((enemy_pieces >> to) & 1) ? MOVE_CAPTURE : 0,
+      );
     }
   }
 }