fixed simple bug