Fixing naming bug

Fixing naming bug