FIX: Remove unnecessary .transpose introduced in PR #984

After the merge of #984, @jni correctly pointed out that moving the `np.newaxis` obviates the need for the `.transpose`. This PR removes the incorrect `.transpose` operation.