fix test and actually work