Fix incorrect assertEquals import