Fix import path in test