Fixed a mistaken bug