Fix robocop warnings