improve tests coverage