minor changes

- fixed a little bug