Fixed a Small Bug