Bug fix add to_s