Fixing bug in flash service test