updated test unit