Modified button tests for TA