Distinguish between ?Int.Lit and ?Int.width

1 files changed, 2 insertions, 2 deletions

diff --git a/src/main/scala/chisel3/util/OneHot.scala b/src/main/scala/chisel3/util/OneHot.scala
index 820c72d6..8a5caf44 100644
--- a/src/main/scala/chisel3/util/OneHot.scala
+++ b/src/main/scala/chisel3/util/OneHot.scala
@@ -41,9 +41,9 @@ object UIntToOH
 {
   def apply(in: UInt, width: Int = -1): UInt =
     if (width == -1) {
-      UInt(1) << in
+      UInt.Lit(1) << in
     } else {
-      (UInt(1) << in(log2Up(width)-1,0))(width-1,0)
+      (UInt.Lit(1) << in(log2Up(width)-1,0))(width-1,0)
     }
 }