fixed broken test :)