Fix flask test.