Fix compile problem

Something slipped the recent merge