Правильно.
В сочетании с количественными числительными возможны варианты: пять грамм и пять граммов. Но без числительных верно: не хватает граммов.
Обе грамматические формы верны.
Им. п. - граммы. Р. п. - граммов и грамм (счетная форма).
Возможны обе формы, в письменной речи лучше второй вариант.
См. в "Искать на Грамоте" на нашем портале.