Yep, I found the same problem. After seeing some gameplay on YT, I found out that the problem was because I had the arm still on me.
Yea, found out that u actually have to put the arm on the puzzle in order to get ur gun back
