Imaging and missing drivers

Trying to apply a custom image to an EliteDesk 800 G3 but when I get to the part where it asks what partition to install it on, the box is empty and says it does not have any storage drivers. I've tried injecting a couple into the image with no success. 

This setup uses an M.2 SSD. Any idea what driver it's actually looking for?

Edit: It's a Gen 3