Fix WK-344