Fix minor error all working