Fix error with Unit Reflection