A small fix for missing imports
By testing initialize_from_db function
initialize_from_db
@hammerd @roscar @danilevc