Removing returns on final assert expectation