Fix rubocop things