Fix k=1 case