Context switching works properly now