From 167c1547966b5aca944494757f51fac1be2d0fe8 Mon Sep 17 00:00:00 2001
From: Mikunj <mikunj@live.com.au>
Date: Tue, 21 May 2019 15:47:19 +1000
Subject: [PATCH] Fix up imports.

---
 SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.h | 5 ++---
 SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.m | 2 --
 2 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.h b/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.h
index bf0378a3b..39e5fe5df 100644
--- a/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.h
+++ b/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.h
@@ -1,9 +1,8 @@
 #import "OWSPrimaryStorage.h"
+#import <AxolotlKit/PreKeyRecord.h>
+#import <AxolotlKit/PreKeyBundle.h>
 #import <YapDatabase/YapDatabase.h>
 
-@class PreKeyRecord;
-@class PreKeyBundle;
-
 NS_ASSUME_NONNULL_BEGIN
 
 @interface OWSPrimaryStorage (Loki)
diff --git a/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.m b/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.m
index 70ee07f30..ff57f5b35 100644
--- a/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.m
+++ b/SignalServiceKit/src/Loki/Crypto/OWSPrimaryStorage+Loki.m
@@ -5,8 +5,6 @@
 #import "OWSDevice.h"
 #import "OWSIdentityManager.h"
 #import "NSDate+OWS.h"
-#import "PreKeyRecord.h"
-#import "PreKeyBundle.h"
 #import "TSAccountManager.h"
 #import "TSPreKeyManager.h"
 #import "YapDatabaseConnection+OWS.h"