Clean up getting flat-field