Minor bug fix to pstpm2