Fix Rubocop tests