Updated so test will pass