Better check for time module...

Signed-off-by: Nick Campbell <nicholas.j.campbell@gmail.com>