Azure VM disconnecting randomly

I'm currently using a Azure VM (Linux) for a school final. The connection keeps dropping every time I try and create a new file.


Output
 

[09:14:06.499] Resolver error: Error: Connecting with SSH timed out
	at g.Timeout (c:\Users\admi\.vscode\extensions\ms-vscode-remote.remote-ssh-0.108.2023110315\out\extension.js:2:642416)
	at Timeout._onTimeout (c:\Users\admi\.vscode\extensions\ms-vscode-remote.remote-ssh-0.108.2023110315\out\extension.js:2:760318)
	at listOnTimeout (node:internal/timers:569:17)
	at process.processTimers (node:internal/timers:512:7)
[09:14:06.504] ------

It initially started when I tried installing mysql as required for the final. Got it installed eventually using a work around, but now I cannot create a .py file without it dropping the connection.

Manage to fix the issue by upgrading the VM, consider this solved.