Add a check for a needed string so that the command works propperly.