Fix failed test due to previous patch