Fix wrong import path in tests