HTC One M9 reboots after pressing the sleep/wake button for the second time

I would press the sleep/wake button once and the screen would turn on then I would do it the second time and the phone would reboot (I'm using the CM12.1 beta2 I know it still has bugs but this also happened with the skydragon ROM)

I'm currently using the android revolution ROM which works perfectly but I really want to use the cm12.1 ROM or any aosp ROM so if anyone could provide a link to a aosp ROM or show me a fix it will be greatly appreciated.