Fix test logic for datetime (#2900)

Fix test logic for datetime (#2900) #3557