fix missed tests