Update tests code so they actually run.