Fix test logic for datetime

Fix test logic for datetime #3555