Fix import for when module invoked as script