fix custom tests