diff --git a/src/main/resources/assets/plutoscoffee/blockstates/coffee_plant.json b/src/main/resources/assets/plutoscoffee/blockstates/coffee_plant.json
index 61806cd..86a3f0f 100644
--- a/src/main/resources/assets/plutoscoffee/blockstates/coffee_plant.json
+++ b/src/main/resources/assets/plutoscoffee/blockstates/coffee_plant.json
@@ -1,19 +1,19 @@
 {
   "variants": {
     "age=0": {
-      "model": "plutoscoffee:block/coffee_stage0"
+      "model": "plutoscoffee:block/coffee_seed"
     },
     "age=1": {
       "model": "plutoscoffee:block/coffee_stage0"
     },
     "age=2": {
-      "model": "plutoscoffee:block/coffee_stage1"
+      "model": "plutoscoffee:block/coffee_stage0"
     },
     "age=3": {
       "model": "plutoscoffee:block/coffee_stage1"
     },
     "age=4": {
-      "model": "plutoscoffee:block/coffee_stage2"
+      "model": "plutoscoffee:block/coffee_stage1"
     },
     "age=5": {
       "model": "plutoscoffee:block/coffee_stage2"
diff --git a/src/main/resources/assets/plutoscoffee/models/block/coffee_seed.json b/src/main/resources/assets/plutoscoffee/models/block/coffee_seed.json
new file mode 100644
index 0000000..e047664
--- /dev/null
+++ b/src/main/resources/assets/plutoscoffee/models/block/coffee_seed.json
@@ -0,0 +1,6 @@
+{
+  "parent": "minecraft:block/crop",
+  "textures": {
+    "crop": "plutoscoffee:blocks/coffee_seed"
+  }
+}
\ No newline at end of file
diff --git a/src/main/resources/assets/plutoscoffee/textures/blocks/coffee_seed.png b/src/main/resources/assets/plutoscoffee/textures/blocks/coffee_seed.png
new file mode 100644
index 0000000..2d5bfc4
Binary files /dev/null and b/src/main/resources/assets/plutoscoffee/textures/blocks/coffee_seed.png differ
diff --git a/src/main/resources/data/plutoscoffee/loot_tables/blocks/coffee_plant.json b/src/main/resources/data/plutoscoffee/loot_tables/blocks/coffee_plant.json
index 8d65497..68cef2d 100644
--- a/src/main/resources/data/plutoscoffee/loot_tables/blocks/coffee_plant.json
+++ b/src/main/resources/data/plutoscoffee/loot_tables/blocks/coffee_plant.json
@@ -20,13 +20,9 @@
           "name": "plutoscoffee:coffee_berry",
           "functions": [
             {
-              "function": "minecraft:apply_bonus",
-              "enchantment": "minecraft:fortune",
-              "formula": "minecraft:binomial_with_bonus_count",
-              "parameters": {
-                "extra": 3,
-                "probability": 0.5714286
-              }
+              "function": "minecraft:set_count",
+              "count": 3,
+              "add": true
             }
           ]
         }
@@ -34,7 +30,7 @@
       "conditions": [
         {
           "condition": "minecraft:block_state_property",
-          "block": "minecraft:potatoes",
+          "block": "plutoscoffee:coffee_plant",
           "properties": {
             "age": "7"
           }