Fix import statements in tests