Fix tests
Merge pull request #99 from korczis/rubocop

Fix tests