diff --git a/packages/subgraph-service/contracts/SubgraphServiceStorage.sol b/packages/subgraph-service/contracts/SubgraphServiceStorage.sol
index cc2bc75b3..bd937815c 100644
--- a/packages/subgraph-service/contracts/SubgraphServiceStorage.sol
+++ b/packages/subgraph-service/contracts/SubgraphServiceStorage.sol
@@ -2,6 +2,8 @@
 pragma solidity 0.8.27;
 
 import { ISubgraphService } from "./interfaces/ISubgraphService.sol";
+import { Allocation } from "./libraries/Allocation.sol";
+import { LegacyAllocation } from "./libraries/LegacyAllocation.sol";
 
 abstract contract SubgraphServiceV1Storage {
     /// @notice Service providers registered in the data service