Improved register in simpleflow