Small fix for parser