Fixed naming bug

Fixed naming bug