Update tests to new default flow state dict