Fix add typing into condition