Fixed broken test build