small fix

I think this works better.