Fix bug introduced in test cases by flash size changes