Fixed a slight bug