fixing wrong condition for Kmu momentum calculation