Add wait for angular