A good fix would be to allow you to enter the design through the shipyard computer. Just adding a button "edit design" and you would be in control of the design.
However, make sure to first let the player leave the station's build mode (if he is in it) before he gets in the design or you'll get weird issues.