fix: updated tests to reflect new inputs