Fix for clock_tests

This is for v1-train!