Adding condition on test
this is probably the fix