Fix silly compile error in WIP code