just fixed some small problem