Fixed a slight bug.