Forgot to update __getitem__