Fixed failing Test