add duration and change test accordingly