Fix wrong time.After