fixing a lil problem