diff --git a/petalinux/axistreamdma/files/axistreamdma.c b/petalinux/axistreamdma/files/axistreamdma.c
index a5a4066..f8f9f40 100755
--- a/petalinux/axistreamdma/files/axistreamdma.c
+++ b/petalinux/axistreamdma/files/axistreamdma.c
@@ -188,14 +188,14 @@ int Rce_Probe(struct platform_device *pdev) {
 
    // Set hardware functions
    // Version 2
-   if ( ((ioread32(dev->reg) >> 24) & 0xFF) >= 2 ) {
+   if ( ((readl(dev->reg) >> 24) & 0xFF) >= 2 ) {
       dev->hwFunc = &(AxisG2_functions);
    }
 
    // Version 1
    else {
-      iowrite32(0x1,((uint8_t *)dev->reg)+0x8);
-      if ( ioread32(((uint8_t *)dev->reg)+0x8) != 0x1 ) {
+      writel(0x1,((uint8_t *)dev->reg)+0x8);
+      if ( readl(((uint8_t *)dev->reg)+0x8) != 0x1 ) {
          release_mem_region(dev->baseAddr, dev->baseSize);
          dev_info(dev->device,"Probe: Empty register space. Exiting\n");
          return(-1);