move code into test