Fixed code style after Rubocop review.