fixing a bug.