fix missed check