Rubocop fixes ;(