Simplify the model, it seems to be working much better