fix test and add pst