This solution should work however I am getting the error message Cannot read property 'getElementById' of null. I believe this may be due to direct DOM access being disabled.

I see the message at the top of the form layout edit window that says

"New client-scripts are run in strict mode, with direct DOM access disabled. "

Even after setting the Isolated script flag I get the same message.