Rubocop fixes