Fix linking bug