Fix the style according to rubocop