Fix rubocop workaround