updated to match test