Update write method on test