Aqua affinity not working!

Two updates ago, my diamond helmet with aqua affinity stopped working. It still gave me armour protection but it didn't stop me from drowning! I really hope this can be fixed in the next MCPE update.