Both times it happened were in the algae water room. When I first entered it after starting the level I walked into the regular water and then couldn't jump to get out. Second time was when I had got to the other side of that room and I was trying to enter that giant tunnel at the end of it but again the jump button stopped working so I couldn't get over that little step. It worked normally again each time after restarting the level, and I loved walking through this so much that I did not mind going through it again at all 😁