fix: fixed bug in initial function