Jupyter + nbwavedrom, JS error

fix.py

# MyHDL + MyHDLPeek + NBWaveDrom
# replace the last <script> tag with a <style onload="">
# vim $PREFIX/lib/python3.7/site-packages/nbwavedrom/__init__.py 

def _draw_wavedrom_javascript(data, width):
    style = ""
    if width != None:
        style = ' style="width: ' + str(int(width)) + 'px'
    htmldata = '<script>' + open(_get_js_path('wavedromskin.js')).read() + '</script>'
    htmldata += '<script>' + open(_get_js_path('wavedrom.min.js')).read() + '</script>'
    htmldata += '<div' + style + '><script type="WaveDrom">' + data + '</script></div>'
    htmldata += '<style onload="WaveDrom.ProcessAll()"></style>'
    return IPython.display.HTML(data=htmldata)