small changes

Just fixed a bug