Fixed a simple bug