No, what they were told is not correct. You are allowed to get apply for SSN after marriage, and plenty of us here have.
I’ll not dispute that it is best practice to apply before, as mix ups like this can happen, but you are well within your rights to apply for SSN as a K-1 after marriage, as long as your I-94 hasn’t yet expired.