Re-add erroneously removed import