Fixed error in link class
Tests passing