fixed a simple bug