Fix the gem

Fixing the gem