Some PSM rendering fixes