diff --git a/src/ItemsByCapacity.java b/src/ItemsByCapacity.java
index faf7f85..7f9ccaf 100644
--- a/src/ItemsByCapacity.java
+++ b/src/ItemsByCapacity.java
@@ -2,15 +2,15 @@
 
 public class ItemsByCapacity {
 	private Capacity capacity;
-	private SiteA siteA;
-	public ItemsByCapacity(Capacity capacity, SiteA siteA) {
+	private SiteWrapper siteWrapper;
+	public ItemsByCapacity(Capacity capacity, SiteWrapper siteWrapper) {
 		this.capacity = capacity;
-		this.siteA = siteA;
+		this.siteWrapper = siteWrapper;
 	}
 	public List<Map<String, Object>> getValue() {
 		List<Map<String, Object>> temp_l1 = new ArrayList<>();
 		{
-			for (Map<String, Object> item: this.siteA.getValue()) {
+			for (Map<String, Object> item: this.siteWrapper.getSiteAValue()) {
 				if ((Integer) item.get("capacity") >= this.capacity.getValue()) {
 					temp_l1.add(item);
 				}