Improve specs
Fix rubocop warning