Switch ModCheck.getCustomBlock to use BlockState