Fixed a bug

Bug was fixed