diff --git a/client/coral-embed-stream/src/Embed.js b/client/coral-embed-stream/src/Embed.js index 1c7898afb..ac9acc931 100644 --- a/client/coral-embed-stream/src/Embed.js +++ b/client/coral-embed-stream/src/Embed.js @@ -151,7 +151,11 @@ class Embed extends Component { } {!loggedIn && } {loggedIn && user && } - + -
+const NewCount = ({commentCount, countCache}) =>
{ countCache && commentCount - countCache > 0 &&
@@ -12,7 +11,8 @@ const NewCount = ({commentCount, countCache}) => NewCount.propTypes = { commentCount: PropTypes.number.isRequired, - countCache: PropTypes.number + countCache: PropTypes.number, + loadMore: PropTypes.func.isRequired }; export default NewCount; diff --git a/client/coral-embed-stream/src/Stream.js b/client/coral-embed-stream/src/Stream.js index d59521484..e68a223ca 100644 --- a/client/coral-embed-stream/src/Stream.js +++ b/client/coral-embed-stream/src/Stream.js @@ -24,7 +24,7 @@ class Stream extends React.Component { componentDidMount() { const {asset, getCounts, updateCountCache} = this.props; - updateCountCache(asset.id, asset.comments.length); + updateCountCache(asset.id, asset.commentCount); // Note: Apollo's built-in polling doesn't work with fetchMore queries, so a // setInterval is being used instead.