Fixing tests for the addition of generate_html