Rewrite this to add the gem.