Fix k function