Fixed import reference issue