Fixed count() and findOne() bug