fix input to pu calculation