Fix warnings in models