fixing tiny bug