Fixed a bug in it