Fixed broken test compilation