fix broken test

Adding missing arguments