Fix yml to run custom tests