Test works properly now