test needs to reflect new return