Fix Push State case in Junit