diff --git a/kubejs/server_scripts/create_picky_wheels/tags.js b/kubejs/server_scripts/create_picky_wheels/tags.js
index 93fa88881..708c1e980 100644
--- a/kubejs/server_scripts/create_picky_wheels/tags.js
+++ b/kubejs/server_scripts/create_picky_wheels/tags.js
@@ -5,8 +5,8 @@ function registerCreatePickyWheelsBiomeTags(event) {
 
 	event.add('createpickywheels:waterwheels_boosted', '#tfc:is_river')
 
-	event.add('createpickywheels:waterwheels_whitelist', '#tfc:is_lake')
-	event.add('createpickywheels:waterwheels_whitelist', '#tfc:is_ocean')
+	event.add('createpickywheels:waterwheels_whitelist', '#tfg:overworld_biomes')
+	event.add('createpickywheels:waterwheels_whitelist', '#tfg:nether_biomes')
 	event.add('createpickywheels:waterwheels_whitelist', '#tfg:mars_biomes')
 	
 	event.add('createpickywheels:windmills_whitelist', '#tfg:overworld_biomes')