It works for me if you click "YES" for the main tax return options as well as all the states that the diagnostic generates for.
