@fix code for 3d part