Added gem guard for easy testing