Refactor test; remove print statement