The drive x 1/2 inch sockets are the right ones for the water heater element. The opening of the sockets is designed to fit the end of the hex. This will allow you to use the correct Torque.
Let's Get It Fixed!
HomeX virtual experts will assess your issue (for free!) and can resolve simple problems.