$J/\psi$ production in Au+Au collisions at $\sqrt{s} = 54.4$ GeV