Forgot to update expected count