Quick fix for #1

This should be a quick and dirty fix for issue #1.  It's not perfect but it gets the job done on my end.