fiexd some stuff in correctness