Plant type "beach" not working properly

I made a plant that I want to only spawn and be plantable in the same places as sugarcane can, so I set its plant type to "beach". But when I went to test my mod, the plant would generate in the same places as a "plains" type plant, and could also be placed by the player in places with no nearby water. How can I fix this?