Fix count for needs one test