diff --git a/build/block.rs b/build/block.rs
index 36df147..e877cf8 100644
--- a/build/block.rs
+++ b/build/block.rs
@@ -511,8 +511,8 @@ pub fn build() -> anyhow::Result<TokenStream> {
             /// If this block is `air`, `cave_air` or `void_air`.
             pub const fn is_air(self) -> bool {
                 matches!(
-                    self.to_kind(),
-                    BlockKind::Air | BlockKind::CaveAir | BlockKind::VoidAir
+                    self,
+                    BlockState::AIR | BlockState::CAVE_AIR | BlockState::VOID_AIR
                 )
             }