fix test

thanks @robertberry