remove debug log + improve first model test