Fix (or at least hope to fix) a test

Signed-off-by: Kwpolska <kwpolska@gmail.com>