Adding bug fixed