How to add a \caption properly?#8

Hello, I wanted to add a caption to my plantuml. I tried to solve this with wrapping inside a figure. But now I have to problem that the plantuml floats sometimes in really bad ways, even sometimes it places itself between other lstlistings :/. Is there a way to do better? ``` \begin{figure}[h] \begin{plantuml} @startuml Alice -> Bob: Authentication Request Bob --> Alice: Authentication Response Alice -> Bob: Another authentication Request Alice <-- Bob: another authentication Response @enduml \end{plantuml} \caption{my caption} \label{my label} \end{figure} ```