fix test following the removing of set_flops.
Merge pull request #1573 from nouiz/fix_test

fix test following the removing of set_flops.