clean imports
add missing method due to change to compiler