Fix PI & PO size bug