minor bug fixe