Adapted the test to the trainer tokens interface