Fix shift and caps behavior