fixing error on test