fix models reference in test and use super to init