Hopefully fixed the compilation error