Fixed an issue with the gem

Signed-off-by: Yuri Tomanek <yuri@tomanek.com.au>