Fix rubocop

@markijbema review