Fixed last broken test