fixing a small problem