Fixes the last test