Updated bug when k=1