fixed a little issue