Small framework fix