Bug fix

Fixing stupid bug