Update to new functional calls

Signed-off-by: Riddhesh Sanghvi <[email protected]>
EasyEngine · Aug 6, 2018 · 5133ce1 · 5133ce1
1 parent 1d6e796
commit 5133ce1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Cron_Command.php b/src/Cron_Command.php
@@ -75,7 +75,7 @@ public function add( $args, $assoc_args ) {
 		EE\Utils\delem_log( 'ee cron add start' );
 
 		if ( ! isset( $args[0] ) || $args[0] !== 'host' ) {
-			$args = EE\Utils\set_site_arg( $args, 'cron' );
+			$args = EE\SiteUtils\auto_site_name( $args, 'cron', __FUNCTION__ );
 		}
 
 		$site     = EE\Utils\remove_trailing_slash( $args[0] );
@@ -219,7 +219,7 @@ public function _list( $args, $assoc_args ) {
 		$all   = EE\Utils\get_flag_value( $assoc_args, 'all' );
 
 		if ( ( ! isset( $args[0] ) || $args[0] !== 'host' ) && ! $all ) {
-			$args = EE\Utils\set_site_arg( $args, 'cron' );
+			$args = EE\SiteUtils\auto_site_name( $args, 'cron', 'list' );
 		}
 
 		if ( isset( $args[0] ) ) {