FIxed a bug