8.4.0 - Image Selection not working on custom module

Trying to create a custom module that includes a couple of fields which are of type “image”. Once deployed I’m not getting the option to select an image, it’s just a fillable text field.

Whats strange is that it was working when I first tested it, then it completed stopped working.