Updated the method being called and tweaked the logic. Signed-off-by: Jamie Phillips <jamie.phillips@suse.com>