fixed a small problem