Fixed a slight issue