Fix add_test