diff --git a/include/linux/xz.h b/include/linux/xz.h
index a3c2b77..d1afab0 100644
--- a/include/linux/xz.h
+++ b/include/linux/xz.h
@@ -262,10 +262,10 @@
 XZ_EXTERN uint32_t xz_crc32(const uint8_t *buf, size_t size, uint32_t crc);
 #endif
 
-STATIC int decompress_unxz(unsigned char *in, long in_size,
-		     long (*fill)(void *dest, unsigned long size),
-		     long (*flush)(void *src, unsigned long size),
-		     unsigned char *out, long *in_used,
+STATIC int decompress_unxz(unsigned char *in, int in_size,
+		     int (*fill)(void *dest, unsigned int size),
+		     int (*flush)(void *src, unsigned int size),
+		     unsigned char *out, int *in_used,
 		     void (*error)(char *x));
 
 #endif
diff --git a/lib/decompress_unxz.c b/lib/decompress_unxz.c
index 0b6bc83..1ddcee3 100644
--- a/lib/decompress_unxz.c
+++ b/lib/decompress_unxz.c
@@ -227,10 +227,10 @@
  * both input and output buffers are available as a single chunk, i.e. when
  * fill() and flush() won't be used.
  */
-STATIC int decompress_unxz(unsigned char *in, long in_size,
-		     long (*fill)(void *dest, unsigned long size),
-		     long (*flush)(void *src, unsigned long size),
-		     unsigned char *out, long *in_used,
+STATIC int decompress_unxz(unsigned char *in, int in_size,
+		     int (*fill)(void *dest, unsigned int size),
+		     int (*flush)(void *src, unsigned int size),
+		     unsigned char *out, int *in_used,
 		     void (*error)(char *x))
 {
 	struct xz_buf b;