Fix compiling. My bad!