Fix style according to rubocop