Fix angular directive & E2E test