Fixed issues with count_ops and assumption