fix tests related to source changes