updated test code after domain object change