Fix Rubocop failures