Fix VAE tests failing when ran from the models/tests directory (#1241)

Fixes #1241