Add correct border color