[#285] Fix missing imports