Make sure to always release the wake lock

b/6700936
b/6809256
b/6778795

Change-Id: Id6fe3bea44884433f6255199d85fa8a1431e44ea