Fixing the test project with changes