fix stupid keyerror