Fixed import guard

Closes #377. The import guard was just out of place.