Fix broken fake test class