Add ability to handle count property correctly in atlas.js