Updating tests to check more conditions