fixing a simple bug