Fix bug - add time import