Wrrote better tests