Fix task edit styling