Fix up testing for flask properly.