Sounds like a div is either missing or out of place.
Can you try the 2nd code I posted above one more time, and if it doesn’t work I can take a closer look to see what’s going on. I have it updated on a local install and it seems to be working on my end.