Update test

refoctoring test file