fix wait for site edit btn

Signed-off-by: Sergey Gaichuk <sergey.gaychuk@gmail.com>