Forgot to update test to reflect class changes