Imports needed for conditional code

Imports aren't resolved for unused conditional code. Whoops.