Minor rubocop fixes