From 76d063f20ab94d6be2463b88861aef737cd79f1e Mon Sep 17 00:00:00 2001
From: Charles Papon <charles.papon.90@gmail.com>
Date: Mon, 24 Feb 2020 22:43:08 +0100
Subject: [PATCH] Fix MulPlugin keep attribute

---
 src/main/scala/vexriscv/plugin/MulPlugin.scala | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/main/scala/vexriscv/plugin/MulPlugin.scala b/src/main/scala/vexriscv/plugin/MulPlugin.scala
index efe3ba0..31714e8 100644
--- a/src/main/scala/vexriscv/plugin/MulPlugin.scala
+++ b/src/main/scala/vexriscv/plugin/MulPlugin.scala
@@ -98,8 +98,8 @@ class MulPlugin(inputBuffer : Boolean = false) extends Plugin[VexRiscv]{
 
       Component.current.afterElaboration{
         //Avoid synthesis tools to retime RS1 RS2 from execute stage to decode stage leading to bad timings (ex : Vivado, even if retiming is disabled)
-        KeepAttribute(input(RS1).getDrivingReg)
-        KeepAttribute(input(RS2).getDrivingReg)
+        KeepAttribute(input(RS1))
+        KeepAttribute(input(RS2))
       }
     }