Updated count function calls test.