Bug fix

I know this is a bad description, not gitty style :))