Fixed the last compilation error