Fixed compilation (accidentally added some wip code).