Fixed for rubocop