diff --git a/docs/src/api.md b/docs/src/api.md
index b049a4f..df3e83c 100644
--- a/docs/src/api.md
+++ b/docs/src/api.md
@@ -598,10 +598,6 @@ passive_state
 
 ## [Pauli](@id API: Pauli)
 
-```@docs
-PauliBasis
-```
-
 ```@docs
 PauliTransferMatrix
 ```
@@ -640,4 +636,4 @@ lazytensor_cachesize
 
 ```@docs
 lazytensor_clear_cache
-```
\ No newline at end of file
+```
diff --git a/test/test_metrics.jl b/test/test_metrics.jl
index 0a27b48..fac005f 100644
--- a/test/test_metrics.jl
+++ b/test/test_metrics.jl
@@ -127,8 +127,7 @@ rho_mix = DenseOperator(rho_ent.basis_l, diagm(ComplexF64[1.0,1.0,1.0,1.0]))
 @test_throws ArgumentError entanglement_entropy(rho_mix, (1,2))
 @test_throws ArgumentError entanglement_entropy(rho_mix, 3)
 
-q2 = PauliBasis(2)
-CNOT = DenseOperator(q2, q2, diagm(0 => [1,1,0,0], 1 => [0,0,1], -1 => [0,0,1]))
+CNOT = dm(spinup(b1))⊗identityoperator(b1) + dm(spindown(b1))⊗sigmax(b1)
 CNOT_sop = SuperOperator(CNOT)
 CNOT_chi = ChiMatrix(CNOT)
 CNOT_ptm = PauliTransferMatrix(CNOT)
diff --git a/test/test_pauli.jl b/test/test_pauli.jl
index ad1109a..d85d7dc 100644
--- a/test/test_pauli.jl
+++ b/test/test_pauli.jl
@@ -5,11 +5,10 @@ using QuantumOpticsBase
 
 @testset "pauli" begin
 
-@test_throws MethodError PauliBasis(1.4)
-
+b = SpinBasis(1//2)
 # Test conversion of unitary matrices to superoperators.
-q2 = PauliBasis(2)
-q3 = PauliBasis(3)
+q2 = b^2
+q3 = b^3
 CZ = DenseOperator(q2, q2, diagm(0 => [1,1,1,-1]))
 CZ_sop = SuperOperator(CZ)