Make key model same broadcasting shape