fix regression caused by the previous change