Fix test loader

Change-Id: I6fd58c6772ba276b898c8eb75822ff29baa9160a