Refactor to test lab with lab