fixing a bug...