Improve coverage for flash functions