Fix wrong repository classes