Fix bad import bug due to folder change