Fix the underwater bug

Fix the underwater bug