Fixed a potential issue