Fixed tests so they are not failing because of me