assertCount() fix