Fix gem spec + added a sample test