Fix wrong import path for tests