Small fix on PSM