fixing a little problem