Bug fix, return string