Fixed tests for deep model