Fix a distribution bug