Fixed a up an issue