added count fix