Fix import datetime issue on badger