Add small distribution fix