Fix rubocop test cases