Fixed imports missed on last commit