Fix buggy test