Fixing tiny issue