FIX failing tests due to the change of the underlying sample generator