Updated so test passes