Repository tests plus bug fixes