Fix rubocop errors