Some fixes to the montage PR

This is the first time I'm attempting something like thing. A PR for a PR...