Fix erorrs after test writing