fixing the broken test...