Rewrite 1 line for test