diff --git a/Tweak.xm b/Tweak.xm
index bb89aba..eeb1732 100644
--- a/Tweak.xm
+++ b/Tweak.xm
@@ -21,19 +21,6 @@
 
 @end
 
-@interface AVPlaybackControlsView
-@property (assign, nonatomic) AVTransportControlsView *transportControlsView;
-@end
-
-@interface AVPlayerViewControllerContentView
-@property (assign, nonatomic) AVPlaybackControlsView *playbackControlsView;
-@end
-
-@interface AVFullScreenViewController
-@property (assign, nonatomic) AVPlayerViewControllerContentView *contentView;
-@end
-
-
 
 
 extern "C" CFNotificationCenterRef CFNotificationCenterGetDistributedCenter(void);
@@ -71,22 +58,6 @@ static void firstUpdate(CFNotificationCenterRef center, void *observer, CFString
 
 }
 
-%hook AVFullScreenViewController
-int test = 30;
--(void)viewDidLayoutSubviews{
-	%orig;
-	if(test > 1){
-		[self.contentView.playbackControlsView.transportControlsView deviceOrientationDidChange];
-		test--;;
-	}
-
-}
-
--(void)viewDidDisappear:(bool)arg1{
-	%orig;
-	test = 10;
-}
-%end
 
 %hook SBOrientationLockManager
 -(SBOrientationLockManager*)init{
@@ -130,6 +101,7 @@ CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
 
 
 %hook AVTransportControlsView
+int test = 30;
 -(AVTransportControlsView *)initWithFrame:(CGRect)frame{
     AVTransportControlsView *origself = %orig(frame);
 	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
@@ -163,9 +135,16 @@ CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
 -(void)dealloc{
     %orig;
     CFNotificationCenterRemoveObserver ( CFNotificationCenterGetDarwinNotifyCenter(), (void*)myObserver, NULL, NULL); 
+	test = 30;
 }
 
-
+-(void)layoutSubviews{
+	%orig;
+	if(test > 1){
+		[self deviceOrientationDidChange];
+		test--;;
+	}
+}
 
 %new
 -(void)deviceOrientationDidChange{
diff --git a/com.gilshahar7.avlock_1.0.1_iphoneos-arm.deb b/com.gilshahar7.avlock_1.0.1_iphoneos-arm.deb
new file mode 100644
index 0000000..3f425d9
Binary files /dev/null and b/com.gilshahar7.avlock_1.0.1_iphoneos-arm.deb differ
diff --git a/control b/control
index ba7a60f..8e1b7cc 100644
--- a/control
+++ b/control
@@ -1,7 +1,7 @@
 Package: com.gilshahar7.avlock
 Name: AVLock
 Depends: mobilesubstrate
-Version: 1.0
+Version: 1.0.1
 Architecture: iphoneos-arm
 Description: Add an orientation lock button to the iOS 11 stock video player
 Maintainer: gilshahar7