Add timeout check