Fix wrong import introduced in 6a1815cd