Fixing Reflection class issue