Fixed a slight problem