diff --git a/client/coral-embed-stream/src/containers/Stream.js b/client/coral-embed-stream/src/containers/Stream.js index b54e69dd6..467242592 100644 --- a/client/coral-embed-stream/src/containers/Stream.js +++ b/client/coral-embed-stream/src/containers/Stream.js @@ -113,7 +113,9 @@ class StreamContainer extends React.Component { componentDidMount() { if (this.props.previousTab) { this.props.data.refetch() - .then(() => this.props.setCommentCountCache(0)); + .then(({data: {asset: {commentCount}}}) => { + return this.props.setCommentCountCache(commentCount); + }); } this.countPoll = setInterval(() => { this.getCounts(this.props.data.variables);