fix assign test case