Adding missin gem import.