Fixed missing datetime import.
Merge pull request #458 from damianavila/fix_dt

Fixed missing datetime import.