Bug fix

Little Bug fix