Small fix for parsing