fixes #4
forgot to add change to main.  fixes #4