Fixed bug in flops counting code