Fixed code final

haha careless