fix findOne method