Week 3 programming assignment error

Why is my code not passing the testcase? Have done everything as asked.

import tensorflow as tf

def compute_total_loss(logits, labels):
Computes the total loss

logits -- output of forward propagation (output of the last LINEAR unit), of shape (6, num_examples)
labels -- "true" labels vector, same shape as Z3

total_loss - Tensor of the total loss value

# Set from_logits to True for categorical cross-entropy
total_loss = tf.reduce_sum(tf.keras.losses.categorical_crossentropy(labels, logits, from_logits=True))

return total_loss

tf.Tensor(0.17102128, shape=(), dtype=float32)
AssertionError: Test does not match. Did you get the reduce sum of your loss functions?

If I recall correctly, you need to also transpose the labels and logits variables.