Final update: I ended up just moving the black wire connection to the screw and left the neutral wire where it was because I didn’t want to mess with something that seemed fine. Filled up the tub and it has been successfully keeping temperature for the past 24 hours! Thank you both for all the help, very happy to get this fixed at no cost