diff --git a/emulated/anisette.py b/emulated/anisette.py
index 54a45ca..0c3254a 100644
--- a/emulated/anisette.py
+++ b/emulated/anisette.py
@@ -5,7 +5,7 @@ from websockets.sync.client import connect
 import json
 from base64 import b64encode
 import random
-import bags
+from pypush import bags
 import requests
 import plistlib
 import icloud.gsa as gsa
@@ -52,7 +52,7 @@ with connect(ANISETTE_SERVER) as websocket:
         elif message["result"] == "GiveEndProvisioningData":
 
             if 'cpim' in message:
-            cpim = message['cpim']
+                cpim = message['cpim']
 
         elif message["result"] == "Timeout":
             break
\ No newline at end of file
diff --git a/proxy/printer.py b/proxy/printer.py
index ca20366..1463555 100644
--- a/proxy/printer.py
+++ b/proxy/printer.py
@@ -106,7 +106,8 @@ def _p_filter(prefix, fields: list[tuple[int, bytes]]):
     )
 
 
-import apns
+from pypush import apns
+
 
 def print_payload(payload: apns.APNSPayload, to_server: bool):
     prefix = "apsd -> APNs" if to_server else "APNs -> apsd"
diff --git a/proxy/proxy.py b/proxy/proxy.py
index 6aa1252..f0a7485 100644
--- a/proxy/proxy.py
+++ b/proxy/proxy.py
@@ -16,7 +16,7 @@ import trio
 from rich.logging import RichHandler
 
 import printer
-import apns
+from pypush import apns
 
 logging.basicConfig(
     level=logging.NOTSET,