Tried added assert argument to test function