Fixed a bug on the software