diff --git a/lib/entities/wake_lock.dart b/lib/entities/wake_lock.dart
index 784f59c85..99acc65ee 100644
--- a/lib/entities/wake_lock.dart
+++ b/lib/entities/wake_lock.dart
@@ -5,7 +5,7 @@ class WakeLock {
 
   Future<void> enableWake() async {
     try {
-      await _utils.invokeMethod<String>('enableWakeScreen');
+      await _utils.invokeMethod<bool>('enableWakeScreen');
     } on PlatformException catch (_) {
       print('Failed enabling screen wakelock');
     }
@@ -13,7 +13,7 @@ class WakeLock {
 
   Future<void> disableWake() async {
     try {
-      await _utils.invokeMethod<String>('disableWakeScreen');
+      await _utils.invokeMethod<bool>('disableWakeScreen');
     } on PlatformException catch (_) {
       print('Failed enabling screen wakelock');
     }