Fix compiling for PMEM