Bug fix

Hmm...