Fix command import