Fixed a slight bug for formatting