Fixed compilation. Not sure if that's correct way, though