fix simple bugs