fix prime test