I suggest that instead of removing the ability to sink, there may be another organelle or adaption : a gas vacuole. It would be an organelle accumulating and regulating light gases to allow its cell to float and control its buoyancy by ejecting and absorbing light gases. The gas vacuole should be inside the membrane, so that it may easily regulate its gases.