fixing a small breakage