zram: Small Cleanup

* Trying to make code look more like upstream

Signed-off-by: Chester Kener <Cl3Kener@gmail.com>