FIX: added missing sizes of the samples