Fix for flash build