Fix missing keys() call