fix for #1

It's less fancy, but it works