Fixing test + adding RepoExtension