fixed a little bug

fixed a little bug