fixing a bug in feature 1