Arreglo propiedades antes de tener hijo

Mi pareja y yo vamos a tener un hijo. Actualmente vivimos en un piso que es de mi propiedad exclusivamente. Aun así, hemos llegado a un acuerdo para pagar ciertos recibos (luz, agua, gas y teléfono) de forma compartida. Mi pareja también tiene un piso de su propiedad exclusivamente que actualmente tiene alquilado.

La cuestión es que me preocupa que, una vez nacido el bebé, nos podamos divorciar/separar. Tal y como tenemos actualmente nuestras propiedades, me da miedo que ante una posible separación mi piso se quede como piso de crianza del niño y que: (a) yo no pueda vivir permanentemente en él, y (b) mi mujer disponga de su piso y parcialmente/totalmente del mio cuando le tenga la custodia del niño.

La pregunta es: ¿Cuál es la mejor opción para arreglar este tema y que ambos estemos protegidos?