1 files changed, 2 insertions, 4 deletions

diff --git a/samples/livepatch/livepatch-shadow-fix2.c b/samples/livepatch/livepatch-shadow-fix2.c
index ff9948f0ec00..d6c62844dc15 100644
--- a/samples/livepatch/livepatch-shadow-fix2.c
+++ b/samples/livepatch/livepatch-shadow-fix2.c
@@ -53,17 +53,15 @@ struct dummy {
 bool livepatch_fix2_dummy_check(struct dummy *d, unsigned long jiffies)
 {
         int *shadow_count;
-        int count;
 
         /*
          * Patch: handle in-flight dummy structures, if they do not
          * already have a SV_COUNTER shadow variable, then attach a
          * new one.
          */
-        count = 0;
         shadow_count = klp_shadow_get_or_alloc(d, SV_COUNTER,
-                                               &count, sizeof(count),
-                                               GFP_NOWAIT);
+                                sizeof(*shadow_count), GFP_NOWAIT,
+                                NULL, NULL);
         if (shadow_count)
                 *shadow_count += 1;