The Royal Marines have been ordered by a military safety watchdog to check the depth of water before disembarkation training after the death of a recruit.