1 files changed, 6 insertions, 8 deletions

diff --git a/drivers/virtio/virtio_balloon.c b/drivers/virtio/virtio_balloon.c
index 1f572c00a1be..c444654fc33f 100644
--- a/drivers/virtio/virtio_balloon.c
+++ b/drivers/virtio/virtio_balloon.c
@@ -275,9 +275,8 @@ static inline s64 towards_target(struct virtio_balloon *vb)
         __le32 v;
         s64 target;
 
-        vb->vdev->config->get(vb->vdev,
-                              offsetof(struct virtio_balloon_config, num_pages),
-                              &v, sizeof(v));
+        virtio_cread(vb->vdev, struct virtio_balloon_config, num_pages, &v);
+
         target = le32_to_cpu(v);
         return target - vb->num_pages;
 }
@@ -286,9 +285,8 @@ static void update_balloon_size(struct virtio_balloon *vb)
 {
         __le32 actual = cpu_to_le32(vb->num_pages);
 
-        vb->vdev->config->set(vb->vdev,
-                              offsetof(struct virtio_balloon_config, actual),
-                              &actual, sizeof(actual));
+        virtio_cwrite(vb->vdev, struct virtio_balloon_config, num_pages,
+                      &actual);
 }
 
 static int balloon(void *_vballoon)
@@ -513,7 +511,7 @@ static void virtballoon_remove(struct virtio_device *vdev)
         kfree(vb);
 }
 
-#ifdef CONFIG_PM
+#ifdef CONFIG_PM_SLEEP
 static int virtballoon_freeze(struct virtio_device *vdev)
 {
         struct virtio_balloon *vb = vdev->priv;
@@ -556,7 +554,7 @@ static struct virtio_driver virtio_balloon_driver = {
         .probe =        virtballoon_probe,
         .remove =       virtballoon_remove,
         .config_changed = virtballoon_changed,
-#ifdef CONFIG_PM
+#ifdef CONFIG_PM_SLEEP
         .freeze =       virtballoon_freeze,
         .restore =      virtballoon_restore,
 #endif