Small fix to regex