fix test

Signed-off-by: Adam Bell-Hanssen <adambell7@gmail.com>